jax

08fcdb72 - [Take 2] Deprecate default Auto AxisType created by make_mesh. From JAX v0.9.0, make_mesh will return `Explicit` AxisType as the default.

[Take 2] Deprecate default Auto AxisType created by make_mesh. From JAX v0.9.0, make_mesh will return `Explicit` AxisType as the default. Co-authored-by: Jake Vanderplas <jakevdp@google.com> Reverts 205126edd2ab3fb1435efb66b4cd5d4fa8d92db3 PiperOrigin-RevId: 829577441