hessians

functorch is JAX-like composable function transforms for PyTorch.

Fast non-allocating calculations of gradients, Jacobians, and Hessians with sparsity support