kazukiosawa / ngd_in_wide_nn Goto Github PK
View Code? Open in Web Editor NEWsimple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-space (by empirical/analytical NTK).
License: MIT License