Convolutions and more as einsum for PyTorch
{KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch
Neural Tangent Kernel (NTK) module for the scikit-learn library
