Commit 55a5e7be authored by Carl Goodrich's avatar Carl Goodrich
Browse files

import vmap

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