jax
d777b52f
- Integrate TorchTPU SDPA with optimized flash attention kernel.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
60 days ago
Integrate TorchTPU SDPA with optimized flash attention kernel. PiperOrigin-RevId: 878086730
References
#34847 - Integrate TorchTPU SDPA with optimized flash attention kernel.
Author
jcc-google
Committer
Google-ML-Automation
Parents
031c1e91
Loading
