jax.lax.gather ============== .. currentmodule:: jax.lax .. autofunction:: gather