jax.numpy.argwhere ================== .. currentmodule:: jax.numpy .. autofunction:: argwhere