Last released Feb 25, 2026
A Jax package for approximate curvature estimation and optimization using KFAC.
Supported by