GitHub repository: JHSchlegel/pytorch-hessian-eigenthings
Primary language: Python
Description: Efficient PyTorch Hessian eigendecomposition tools!