Code samples

Computing GGN eigenvalues

Computing GGN eigenvalues

Computing GGN eigenpairs

Computing GGN eigenpairs

Computing directionally damped Newton steps

Computing directionally damped Newton steps

Computing directional derivatives along GGN eigenvectors

Computing directional derivatives along GGN eigenvectors

Computing empirical NTKs

Computing empirical NTKs