There are no reviews yet. Be the first to send feedback to the community and the maintainers!
Repository Details
simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-space (by empirical/analytical NTK).