flax
e0de630d - Remove uses of JAX-internal test utilites from Flax.

Commit

3 years ago

Remove uses of JAX-internal test utilites from Flax. PiperOrigin-RevId: 481203353

Author

hawkinsp

Committer

a-googler

Parents