/pytorch-hessian-eigenthings

Efficient PyTorch Hessian eigendecomposition tools!

Primary LanguagePythonMIT LicenseMIT

Stargazers