jax.hessian =========== .. currentmodule:: jax .. autofunction:: hessian