jax.ops.segment\_max
====================

.. currentmodule:: jax.ops

.. autofunction:: segment_max