flax
c25f546f - Merge pull request #3691 from chiamp:nnx_rms

Commit

2 years ago

Merge pull request #3691 from chiamp:nnx_rms PiperOrigin-RevId: 606749684

References

#3691 - added nnx rmsnorm

Author

a-googler

a-googler

Parents

Loading