Default `jax.Array` to `True` globally. See https://jax.readthedocs.io/en/latest/jax_array_migration.html for migration to jax.Array.
Default `jax.Array` to `True` globally. See https://jax.readthedocs.io/en/latest/jax_array_migration.html for migration to jax.Array.

PiperOrigin-RevId: 488764287

google/jax,google/jax,google/jax,google/jax