jax
80784a5b
- Merge pull request #26387 from Rifur13:sharding
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Merge pull request #26387 from Rifur13:sharding PiperOrigin-RevId: 738919611
References
#26387 - Add sharding support for dynamic masks in TPU splash attention
Author
Google-ML-Automation
Parents
ea7fa29b
412f1d35
Loading
