jax.experimental.pallas.atomic\_max
===================================

.. currentmodule:: jax.experimental.pallas

.. autofunction:: atomic_max