jax
6bc1cdb7
- Move the pmap API from api.py to pmap.py to avoid local imports
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
234 days ago
Move the pmap API from api.py to pmap.py to avoid local imports PiperOrigin-RevId: 886490494
References
#36065 - Move the pmap API from api.py to pmap.py to avoid local imports
Author
yashk2810
Committer
Google-ML-Automation
Parents
d1eb0fb0
Loading
