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

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

.. autoclass:: SwizzleTransform

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~SwizzleTransform.__init__
      ~SwizzleTransform.batch
      ~SwizzleTransform.to_gpu_transform
      ~SwizzleTransform.undo
      ~SwizzleTransform.undo_to_gpu_transform
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~SwizzleTransform.swizzle