kfac-jax
A Jax package for approximate curvature estimation and optimization using KFAC.
Installation
To install this package, run one of the following:
Usage Tracking
0.0.7
0.0.6
0.0.5
3 / 8 versions selected
Downloads (Last 6 months): 0
About
Summary
A Jax package for approximate curvature estimation and optimization using KFAC.
Last Updated
May 21, 2025 at 04:07
License
Apache-2.0
Total Downloads
5.2K
Supported Platforms
noarch