jax
d76705da
- [Pallas/TPU] Add vmap support for dynamic grid
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[Pallas/TPU] Add vmap support for dynamic grid PiperOrigin-RevId: 603502393
References
#19632 - [Pallas/TPU] Add vmap support for dynamic grid
Author
sharadmv
Committer
a-googler
Parents
b6cda052
Loading