/pytorch-hessian-eigenthings

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

Primary LanguagePythonMIT LicenseMIT

Watchers