jax
1f6152d1
- [Pallas] Use Pallas cost estimator for flash attention.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[Pallas] Use Pallas cost estimator for flash attention. PiperOrigin-RevId: 698573265
References
#148 - CI: 11/22/24 upstream sync
Author
justinjfu
Committer
Google-ML-Automation
Parents
f39392ea
Loading