Second Order Optimization Python Packages

pyhessian

PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks

2K 789 127

kron-torch

An implementation of PSGD Kron second-order optimizer for PyTorch

2K 102 6

torchzero

Modular optimization library for PyTorch (work-in-progress).

2K 13 0

zij

A canon of deep learning optimizers.

927 5 0

psgd-jax

An implementation of PSGD optimizer in JAX.

634 36 4

optami

Second-Order PyTorch Optimizers

208 30 8

python-somax

Second-Order Methods in JAX

156 7 1