jax
412f1d35
- Adding sharding support to dynamic masks
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Adding sharding support to dynamic masks
References
#26387 - Add sharding support for dynamic masks in TPU splash attention
Author
Rifur13
Committer
Rifur13
Parents
59e480db
Loading
