nngeometry (pypi): Manipulate geometry matrices in Pytorch
Keywords: ekfac|fim|fisher-information-matrix|kfac|neural-tangent-kernel|ntk|pytorch