Commit 57538da6 authored by Maximilian LECHNER's avatar Maximilian LECHNER
Browse files

Changed tooltip of run_minimization_while_neighbor_list. It is now...

Changed tooltip of run_minimization_while_neighbor_list. It is now differentiable due to a change in jax-md.
parent a1c2ebfd
......@@ -78,9 +78,8 @@ def run_minimization_scan_neighbor_list(
return state.position, get_maxgrad(state), nbrs
# This version is internally jitted and not differentiable
# The benefit is that it terminates when properly minimized
# Always use this version unless you need AD
# This version is internally jitted, differentiable and uses neighbor lists.
# The benefit is that it terminates when properly minimized.
def run_minimization_while_neighbor_list(
energy_fn,
neighbor_fn,
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment