mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-14 10:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add performance note at the top of sparse docs

Browse Source
This commit is contained in:
Jake VanderPlas 2025-01-17 10:17:16 -08:00
parent a4a657bc43
commit 3141453b1b
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docs/jax.experimental.sparse.rst
View File

@ -1,6 +1,12 @@
``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