https://keroseneheaterreview.blogspot.com/2023/09/pytorch-most-efficient-jacobianhessian.html