jax.experimental.pallas.GridSpec
================================

.. currentmodule:: jax.experimental.pallas

.. autoclass:: GridSpec

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~GridSpec.__init__
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~GridSpec.scratch_shapes
      ~GridSpec.grid
      ~GridSpec.grid_names
      ~GridSpec.in_specs
      ~GridSpec.out_specs