Update code to be compatible with jax 0.4.38

mavenlin released this 17 Jan 06:31

· 2 commits to main since this release

b14ede2

fix calls to WrappedFun and lu.transformation
add hash to Tracer
add to_tangent_aval

Assets 2