jax.experimental.pallas.BlockSpec
=================================

.. currentmodule:: jax.experimental.pallas

.. autoclass:: BlockSpec

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~BlockSpec.__init__
      ~BlockSpec.to_block_mapping
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~BlockSpec.block_shape
      ~BlockSpec.index_map
      ~BlockSpec.indexing_mode
      ~BlockSpec.memory_space