jax.lax.psum\_scatter
=====================

.. currentmodule:: jax.lax

.. autofunction:: psum_scatter