Merge pull request #33922 from jakevdp:sparse-note

PiperOrigin-RevId: 844853575

Author: Google-ML-Automation

docs/jax.experimental.sparse.rst

@@ -1,12 +1,6 @@
 ``jax.experimental.sparse`` module
 ==================================
 
-.. note::
-
-   The methods in ``jax.experimental.sparse`` are experimental reference
-   implementations, and not recommended for use in performance-critical
-   applications.
-
 .. automodule:: jax.experimental.sparse
 
 .. currentmodule:: jax.experimental.sparse

jax/experimental/sparse/__init__.py

@@ -15,10 +15,17 @@
 """
 .. currentmodule:: jax.experimental.sparse
 
+.. note::
+
+  The methods in ``jax.experimental.sparse`` are experimental reference implementations,
+  and not recommended for use in performance-critical applications. The submodule is no
+  longer being actively developed, but the team will continue supporting existing features
+  as best we can.
+
 The :mod:`jax.experimental.sparse` module includes experimental support for sparse matrix
-operations in JAX. It is under active development, and the API is subject to change. The
-primary interfaces made available are the :class:`BCOO` sparse array type, and the
-:func:`sparsify` transform.
+operations in JAX. The primary interfaces made available are the :class:`BCOO` sparse array
+type, and the :func:`sparsify` transform.
+
 
 Batched-coordinate (BCOO) sparse matrices
 -----------------------------------------