Ntk Python Packages

einconv

Convolutions and more as einsum for PyTorch

nngeometry

{KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch

scikit-ntk

Implementation of the neural tangent kernel for scikit-learn's Gaussian process module.