jax
da5e7bbc
- [Pallas TPU] Add support for VectorSubcore + TensorCore async MPMD
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
144 days ago
[Pallas TPU] Add support for VectorSubcore + TensorCore async MPMD PiperOrigin-RevId: 908912100
References
#37326 - [Pallas TPU] Add support for VectorSubcore + TensorCore async MPMD
Author
sharadmv
Committer
Google-ML-Automation
Parents
8f2e8fb2
Loading
