
jax.ops package
=================

.. currentmodule:: jax.ops

.. automodule:: jax.ops


Indexed update operators
------------------------

JAX is intended to be used with a functional style of programming, and hence
does not support NumPy-style indexed assignment directly.

.. autosummary::
  :toctree: _autosummary

    index
    index_update
    index_add
    segment_sum


jax.ops package
=================

.. currentmodule:: jax.ops

.. automodule:: jax.ops


Indexed update operators
------------------------

JAX is intended to be used with a functional style of programming, and hence
does not support NumPy-style indexed assignment directly.

.. autosummary::
  :toctree: _autosummary

    index
    index_update
    index_add
    index_min
    index_max
    segment_sum

Update docs to include index_{min,max}.
Update docs to include index_{min,max}.
