jax
833c7ba7
- Allow generation of sharding strategies with mixed mesh shapes by default.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Allow generation of sharding strategies with mixed mesh shapes by default. PiperOrigin-RevId: 641930205
References
#21771 - Allow generation of sharding strategies with mixed mesh shapes by default.
Author
Google-ML-Automation
Committer
a-googler
Parents
0739d520
Loading