Skip to content

fix(jax): keep parameterization cache off ModelInstance + auto-register pytrees #138

fix(jax): keep parameterization cache off ModelInstance + auto-register pytrees

fix(jax): keep parameterization cache off ModelInstance + auto-register pytrees #138