flax
2d47c931 - linen_meta_test: make test compatible with JAX custom prng

Commit

2 years ago

linen_meta_test: make test compatible with JAX custom prng Part of https://github.com/google/jax/issues/9263 PiperOrigin-RevId: 528004952

Author

Jake VanderPlas

Committer

a-googler

Parents