https://github.com/google/jax
Raw File
Tip revision: 0118f8d568d4d94665e01f284f63f0a5d9865f5e authored by Yash Katariya on 08 December 2022, 00:01:43 UTC
Prepare for jax and jaxlib 0.4.0 release
Tip revision: 0118f8d
jax.experimental.sparse.rst
jax.experimental.sparse module
==============================

.. automodule:: jax.experimental.sparse

API
---

.. autosummary::
   :toctree: _autosummary

   BCOO
   sparsify
   bcoo_broadcast_in_dim
   bcoo_concatenate
   bcoo_dot_general
   bcoo_dot_general_sampled
   bcoo_extract
   bcoo_fromdense
   bcoo_multiply_dense
   bcoo_multiply_sparse
   bcoo_reduce_sum
   bcoo_reshape
   bcoo_sort_indices
   bcoo_sum_duplicates
   bcoo_todense
   bcoo_transpose
back to top