jax.experimental.pallas.atomic\_xchg
====================================

.. currentmodule:: jax.experimental.pallas

.. autofunction:: atomic_xchg