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