jax.experimental.sparse.JAXSparse
=================================

.. currentmodule:: jax.experimental.sparse

.. autoclass:: JAXSparse

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~JAXSparse.__init__
      ~JAXSparse.block_until_ready
      ~JAXSparse.sum
      ~JAXSparse.transpose
      ~JAXSparse.tree_flatten
      ~JAXSparse.tree_unflatten
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~JAXSparse.T
      ~JAXSparse.ndim
      ~JAXSparse.size
      ~JAXSparse.data
      ~JAXSparse.shape
      ~JAXSparse.nse
      ~JAXSparse.dtype