f-dangel / hbp

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

Home Page:https://arxiv.org/abs/1902.01813

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

f-dangel/hbp Stargazers