Ppytorch-hessian-eigenthings
github.com ↗Efficient PyTorch Hessian eigendecomposition using the Hessian-vector product and stochastic power iteration.
READMEView on GitHub ↗
PPEfficient PyTorch Hessian eigendecomposition using the Hessian-vector product and stochastic power iteration.