kfac-jax
A Jax package for approximate curvature estimation and optimization using KFAC.
A Jax package for approximate curvature estimation and optimization using KFAC.
To install this package, run one of the following:
Summary
A Jax package for approximate curvature estimation and optimization using KFAC.
Last Updated
Feb 25, 2026 at 06:19
License
Apache-2.0
Total Downloads
5.7K
Version Downloads
52
Supported Platforms