ltatzel / PyTorchHessianFree Star 30 Code Issues Pull requests PyTorch implementation of the Hessian-free optimizer deep-neural-networks deep-learning conjugate-gradient optimization-algorithms stochastic-optimizers newtons-method matrix-free second-order-optimization hessian-free pytorch-implementation ggn Updated Jun 14, 2024 Python
f-dangel / curvlinops Star 17 Code Issues Pull requests scipy linear operators for the Hessian, Fisher/GGN, and more in PyTorch fisher pytorch linear scipy operators hessian linalg ggn linearoperator Updated Sep 23, 2024 Python
f-dangel / einconv Star 12 Code Issues Pull requests Convolutions and more as einsum for PyTorch fisher pytorch convolution ntk transpose-convolution im2col kfac einsum tensornetwork einsum-notation vjp jvp ggn Updated Jun 6, 2024 Python