scipy Linear operators for curvature matrices in PyTorch
conda install conda-forge::curvlinops-for-pytorch