``jax.test_util`` module
========================

.. currentmodule:: jax.test_util

.. automodule:: jax.test_util

List of Functions
-----------------

.. autosummary::
   :toctree: _autosummary

   check_grads
   check_jvp
   check_vjp