PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks
Stochastic Second-Order Methods in JAX