jax
2420937e - Remove PositionalSharding from JAX now that 0.6.2 is out and next release is 0.7.0.

Commit
199 days ago
Remove PositionalSharding from JAX now that 0.6.2 is out and next release is 0.7.0. Also remove `GSPMDSharding` export from `jax.sharding` endpoint PiperOrigin-RevId: 773021473
Author
Committer
Parents
Loading