flax

045df4a3 - The default AxisType returned by `make_mesh` is going to change to `AxisType.Explicit` in a follow up CL. This change makes it so that the current code falls back to the old behavior.

The default AxisType returned by `make_mesh` is going to change to `AxisType.Explicit` in a follow up CL. This change makes it so that the current code falls back to the old behavior. This is a trivial and safe change without any change in behavior. No functional changes intended It is highly encouraged to transition your codebase to explicit sharding mode and replace `AxisType.Auto` -> `AxisType.Explicit`. See https://docs.jax.dev/en/latest/notebooks/explicit-sharding.html for more information. Note that this CL was generated via `refex` and sent via rosie. PiperOrigin-RevId: 829569991