https://mingredbestcheapprice.blogspot.com/2023/02/pytorch-most-efficient-jacobianhessian.html