flax
54fa32d9 - Future-proof imports of jax.new_ref / jax.Ref.

Commit

77 days ago

Future-proof imports of jax.new_ref / jax.Ref. Old aliases will be deprecated starting in JAX v0.8.0. PiperOrigin-RevId: 812911822

References

Author

Jake VanderPlas

Committer

a-googler

Parents