jax.numpy.index\_exp ==================== .. currentmodule:: jax.numpy .. autodata:: index_exp