Commit dabeb4b4 authored by Carl Goodrich's avatar Carl Goodrich
Browse files

add import

parent 14e80fc0
import jax.numpy as jnp
from jax_md import space
def _remove_rattlers_oneshot(R, bonds, node_arrays, bond_arrays):
......
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