jax
40ce44d1 - Add `ShardingTypeError` to all sharding rules in JAX

Commit

326 days ago

Add `ShardingTypeError` to all sharding rules in JAX PiperOrigin-RevId: 739205830

References

Author

yashk2810

Committer

Google-ML-Automation

Parents