jax
37cd914c
- JAX changes to support unreduced + partial manual shard_map.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
16 days ago
JAX changes to support unreduced + partial manual shard_map. PiperOrigin-RevId: 988342781
References
#40957 - JAX changes to support unreduced + partial manual shard_map.
Author
yashk2810
Committer
Google-ML-Automation
Parents
4995a7b9
Loading
