jax
527aef3a - Added a slow (but working!) implementation of layer norm in Pallas via Mosaic GPU
Commit
1 year ago
References
Author
Committer
Parents