hessian-free
PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks
Composable Second-Order Optimization for JAX and Optax.