GitHub repository: mtkwT/PyHessian
Description: PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks