jax.ops.index\_max
==================

.. currentmodule:: jax.ops

.. autofunction:: index_max