jax.numpy.mask\_indices ======================= .. currentmodule:: jax.numpy .. autofunction:: mask_indices