jax.experimental.pallas.mosaic\_gpu.as\_torch\_kernel
=====================================================

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

.. autofunction:: as_torch_kernel