jax.experimental.pallas.with\_scoped
====================================

.. currentmodule:: jax.experimental.pallas

.. autofunction:: with_scoped