piwheels

nngeometry

{KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install nngeometry

Dependencies

• torch

Releases

Version	Released	Bookworm Python 3.11	Trixie Python 3.13	Files
0.4	2025-04-13
0.3.2	2025-04-12
0.3	2023-05-23
nngeometry-0.3-py3-none-any.whl (26 KB)
0.2.1	2021-02-24
nngeometry-0.2.1-py3-none-any.whl (21 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue