neural-tangent-kernel
{KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch
Neural Tangent Kernel (NTK) module for the scikit-learn library
A Python library implementing the DKS/TAT neural network transformation method.