flax
fdf877cc - Make flax tests compatible with jax.enable_custom_prng

Commit

2 years ago

Make flax tests compatible with jax.enable_custom_prng PiperOrigin-RevId: 527400813

Author

Jake VanderPlas

Committer

a-googler

Parents