VPanjeta / pytorch-hessian-eigenthings

Efficient PyTorch Hessian eigendecomposition using the Hessian-vector product and stochastic power iteration

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

VPanjeta/pytorch-hessian-eigenthings Watchers