jax
7904b86b - Remove `with_spec` from NamedSharding and replace with `.update`

Commit

344 days ago

Remove `with_spec` from NamedSharding and replace with `.update` PiperOrigin-RevId: 771244288

References

Author

yashk2810

Committer

Google-ML-Automation

Parents