jax.lax.collapse
================

.. currentmodule:: jax.lax

.. autofunction:: collapse