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

github.com/wangleiphy/kfac-jax

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.