Last released Apr 4, 2024
A Jax package for approximate curvature estimation and optimization using KFAC.
Supported by