jax.test_util
module#
List of Functions#
|
Check gradients from automatic differentiation against finite differences. |
|
Check a JVP from automatic differentiation against finite differences. |
|
Check a VJP from automatic differentiation against finite differences. |