Temporarily disable integer index check in jnp.take_along_axis.
Temporarily disable integer index check in jnp.take_along_axis.

This check broke some JAX users; disable it to give time to fix them.

PiperOrigin-RevId: 441993808

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