Commit 2385c439 authored by Maximilian HUEBL's avatar Maximilian HUEBL
Browse files

Added missing imports to

parent 7cb55a0b
import jax.numpy as jnp
from jax import vmap
from jax_md import space
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