jax.experimental.pallas.MemoryRef
=================================

.. currentmodule:: jax.experimental.pallas

.. autoclass:: MemoryRef

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~MemoryRef.__init__
      ~MemoryRef.get_array_aval
      ~MemoryRef.get_ref_aval
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~MemoryRef.shape
      ~MemoryRef.dtype
      ~MemoryRef.memory_space