Neural Tangent Kernel Python Packages

nngeometry

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

scikit-ntk

Neural Tangent Kernel (NTK) module for the scikit-learn library

dks

A Python library implementing the DKS/TAT neural network transformation method.