Backend Integrations
====================

This module contains integrations with external backends like CuPy and JAX.

CuPy Backend
------------

.. currentmodule:: torchsparsegradutils.cupy

.. automodule:: torchsparsegradutils.cupy.cupy_sparse_solve
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: torchsparsegradutils.cupy.cupy_bindings
   :members:
   :undoc-members:
   :show-inheritance:

JAX Backend
-----------

.. currentmodule:: torchsparsegradutils.jax

.. automodule:: torchsparsegradutils.jax.jax_sparse_solve
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: torchsparsegradutils.jax.jax_bindings
   :members:
   :undoc-members:
   :show-inheritance: