/hbp

Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations

Primary LanguagePythonMIT LicenseMIT

Stargazers