jax
2ed3944c - Replace `SDS(vma: frozenset = ...)` to `SDS(manual_type: jax.sharding.ManualAxisType = ...)`. Also expose `jax.sharding.ManualAxisType` as a public API.

Commit

52 days ago

Replace `SDS(vma: frozenset = ...)` to `SDS(manual_type: jax.sharding.ManualAxisType = ...)`. Also expose `jax.sharding.ManualAxisType` as a public API. PiperOrigin-RevId: 885616303

Author

yashk2810

Committer

Google-ML-Automation

Parents

93772b38

jax 2ed3944c - Replace `SDS(vma: frozenset = ...)` to `SDS(manual_type: jax.sharding.ManualAxisType = ...)`. Also expose `jax.sharding.ManualAxisType` as a public API.

jax
2ed3944c - Replace `SDS(vma: frozenset = ...)` to `SDS(manual_type: jax.sharding.ManualAxisType = ...)`. Also expose `jax.sharding.ManualAxisType` as a public API.