jax
24ad445e
- [Pallas] Add support for pytrees in scalar prefetch
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
[Pallas] Add support for pytrees in scalar prefetch PiperOrigin-RevId: 570453699
References
#17822 - [Pallas] Add support for pytrees in scalar prefetch
Author
sharadmv
Committer
a-googler
Parents
ee8af098
Loading
