jax.ops.index ============= .. currentmodule:: jax.ops .. autodata:: index