https://travelebookdownload.blogspot.com/2024/08/pytorch-most-efficient-jacobianhessian.html