jax.lax.index\_take
===================

.. currentmodule:: jax.lax

.. autofunction:: index_take