carlonicolini / nngeometry Goto Github PK
View Code? Open in Web Editor NEWThis project forked from tfjgeorge/nngeometry
{KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch
Home Page: https://nngeometry.readthedocs.io
License: MIT License