GitHub repository: tfjgeorge/nngeometry
Primary language: Python
Description: {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch