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

Add jax.test_util to public API docs

Browse Source
This commit is contained in:
Stephan Hoyer 2025-01-23 16:04:35 -08:00
parent 1d016962c5
commit 458f6a6efe
2 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/jax.rst
View File

@ -25,6 +25,7 @@ Subpackages
jax.ops
jax.profiler
jax.stages
jax.test_util
jax.tree
jax.tree_util
jax.typing

16
docs/jax.test_util.rst Normal file
View File

@ -0,0 +1,16 @@
``jax.test_util`` module
===================
.. currentmodule:: jax.test_util
.. automodule:: jax.test_util
List of Functions
-----------------
.. autosummary::
:toctree: _autosummary
check_grads
check_jvp
check_vjp