Skip to content

rewrite _make_jaxpr() function to be compatible with jax==0.4.29 #16

rewrite _make_jaxpr() function to be compatible with jax==0.4.29

rewrite _make_jaxpr() function to be compatible with jax==0.4.29 #16