jax
527aef3a
- Added a slow (but working!) implementation of layer norm in Pallas via Mosaic GPU
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Added a slow (but working!) implementation of layer norm in Pallas via Mosaic GPU PiperOrigin-RevId: 634710243
References
#21275 - Added a slow (but working!) implementation of layer norm in Pallas via Mosaic GPU
Author
superbobry
Committer
a-googler
Parents
5e2710c2
Loading