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

lattice quick fix

parent 077c3b1b
import jax.numpy as jnp
from jax.api import jit, vmap
from jax import jit, vmap
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