jax
f9f97f7f
- Merge pull request #4835 from google:xmap
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
Merge pull request #4835 from google:xmap PiperOrigin-RevId: 344284665
References
#4835 - simplify the implementation of vmap collectives
Author
a-googler
Parents
5ac16d14
ebd51e12
Loading
