jax.jacrev ========== .. currentmodule:: jax .. autofunction:: jacrev