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

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

.. autoclass:: GPUBlockSpec

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~GPUBlockSpec.__init__
      ~GPUBlockSpec.to_block_mapping
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~GPUBlockSpec.block_shape
      ~GPUBlockSpec.index_map
      ~GPUBlockSpec.indexing_mode
      ~GPUBlockSpec.memory_space
      ~GPUBlockSpec.transforms