jax.lax.pshuffle
================

.. currentmodule:: jax.lax

.. autofunction:: pshuffle