jax.test_util module

.. currentmodule:: jax.test_util

.. automodule:: jax.test_util

List of Functions

.. autosummary:: :toctree: _autosummary

check_grads check_jvp check_vjp