mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +00:00
17 lines
238 B
ReStructuredText
17 lines
238 B
ReStructuredText
``jax.test_util`` module
|
|
========================
|
|
|
|
.. currentmodule:: jax.test_util
|
|
|
|
.. automodule:: jax.test_util
|
|
|
|
List of Functions
|
|
-----------------
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
check_grads
|
|
check_jvp
|
|
check_vjp
|