flax
816cc5e9 - Merge pull request #3406 from chiamp:rmsnorm

Commit
2 years ago
Merge pull request #3406 from chiamp:rmsnorm PiperOrigin-RevId: 572295068
Author
Loading