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

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

.. autoclass:: MemorySpace

   
   .. automethod:: __init__

   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~MemorySpace.GMEM
      ~MemorySpace.SMEM
      ~MemorySpace.TMEM
      ~MemorySpace.REGS