flax
45ce6ee2 - Merge pull request #4749 from vfdev-5:mha-add-keep-rngs-arg

Commit

258 days ago

Merge pull request #4749 from vfdev-5:mha-add-keep-rngs-arg PiperOrigin-RevId: 759305275

References

Author

a-googler

Parents