jax.experimental.pallas.mosaic\_gpu.kernel
==========================================

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

.. autofunction:: kernel