jax
d76705da - [Pallas/TPU] Add vmap support for dynamic grid
Commit
View On GitHub
Commit
1 year ago
[Pallas/TPU] Add vmap support for dynamic grid PiperOrigin-RevId: 603502393
Author
sharadmv sharadmv
Committer
a-googler a-googler
Parents
b6cda052
FAQ Terms Privacy Refunds Impressum
Loading