jax.experimental.pallas.tpu.MemorySpace
=======================================

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

.. autoclass:: MemorySpace

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~MemorySpace.from_type
      ~MemorySpace.like
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~MemorySpace.VMEM
      ~MemorySpace.VMEM_SHARED
      ~MemorySpace.SMEM
      ~MemorySpace.CMEM
      ~MemorySpace.SEMAPHORE
      ~MemorySpace.HBM