milpan / nngeometry Goto Github PK
View Code? Open in Web Editor NEWThis project forked from tfjgeorge/nngeometry
NNGeometry is a PyTorch library for computing Fisher Information Matrices and Neural Tangent Kernels
Home Page: https://nngeometry.readthedocs.io
License: MIT License