jax.Array.argmax
================

.. currentmodule:: jax

.. automethod:: Array.argmax