flax
6f2b08e0 - Fix instances of JAX key reuse in flax tests

Commit

1 year ago

Fix instances of JAX key reuse in flax tests PiperOrigin-RevId: 614703874

Author

Jake VanderPlas

Committer

a-googler

a-googler

Parents

Loading