jax.experimental.pallas.tpu.PrefetchScalarGridSpec
==================================================

.. currentmodule:: jax.experimental.pallas.tpu

.. autoclass:: PrefetchScalarGridSpec

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~PrefetchScalarGridSpec.__init__
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~PrefetchScalarGridSpec.scratch_shapes
      ~PrefetchScalarGridSpec.num_scalar_prefetch
      ~PrefetchScalarGridSpec.grid
      ~PrefetchScalarGridSpec.grid_names
      ~PrefetchScalarGridSpec.in_specs
      ~PrefetchScalarGridSpec.out_specs