flax
03f685a9 - Fix abstract PRNG shape when creating params.

Commit

4 years ago

Fix abstract PRNG shape when creating params. PiperOrigin-RevId: 421944218

Author

jpuigcerver

jpuigcerver

Committer

a-googler

a-googler

Parents

Loading