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

Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models (and their initializations) to make them easier to train.