kazukiosawa / ngd_in_wide_nn

simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-space (by empirical/analytical NTK).
14Updated 4 years ago

Alternatives and similar repositories for ngd_in_wide_nn:

Users that are interested in ngd_in_wide_nn are comparing it to the libraries listed below