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).

github.com/kazukiosawa/ngd_in_wide_nn
by @kazukiosawa

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.