kfac-jax

Python ★ 327 updated 10d ago

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

No plain-English explanation yet — one is being written right now. Check back in a minute.