Skip to content

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

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

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