flax
d33a33e6 - Merge pull request #3354 from chiamp:dunder

Commit

2 years ago

Merge pull request #3354 from chiamp:dunder PiperOrigin-RevId: 567429834

References

#3354 - use jax.Array type for rng keys

Author

a-googler

a-googler

Parents

Loading