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