Welcome to NNGeometry’s documentation!

NNGeometry is a library built on top of PyTorch aiming at giving tools to easily manipulate and study properties of Fisher Information Matrices and tangent kernels.

You can start by looking at the quick start example below. Convinced? Then install NNGeometry, try the tutorials or explore the API reference.

Warning

NNGeometry is under developement, as such it is possible that core components change when between versions.

Tutorials

Indices and tables

Quick start