jax.experimental.pallas.mosaic\_gpu.set\_max\_registers
=======================================================

.. currentmodule:: jax.experimental.pallas.mosaic_gpu

.. autofunction:: set_max_registers