jax
584f9bfb - Make `axis_types` argument to `Mesh`, `AbstractMesh` and `make_mesh` an optional positional argument. It can still be used as a keyword argument so it's backwards compatible.

Commit

1 year ago

Make `axis_types` argument to `Mesh`, `AbstractMesh` and `make_mesh` an optional positional argument. It can still be used as a keyword argument so it's backwards compatible. PiperOrigin-RevId: 774313420

References

#29634 - Make `axis_types` argument to `Mesh`, `AbstractMesh` and `make_mesh` an optional positional argument. It can still be used as a keyword argument so it's backwards compatible.

Author

yashk2810

Committer

Google-ML-Automation

Parents

53ff15a3

jax 584f9bfb - Make `axis_types` argument to `Mesh`, `AbstractMesh` and `make_mesh` an optional positional argument. It can still be used as a keyword argument so it's backwards compatible.

jax
584f9bfb - Make `axis_types` argument to `Mesh`, `AbstractMesh` and `make_mesh` an optional positional argument. It can still be used as a keyword argument so it's backwards compatible.