Convolutions and more as einsum for PyTorch
PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
