kfac-jax

A Jax package for approximate curvature estimation and optimization using KFAC.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install kfac-jax

Releases

Version Released Bullseye
Python 3.9
Bookworm
Python 3.11
Files
0.0.7 2025-05-20
0.0.6 2024-04-04    
0.0.5 2023-05-16  
0.0.3 2022-09-23  
0.0.2 2022-06-07  
0.0.1 2022-04-01  

Issues with this package?

Page last updated 2025-06-27 20:31:30 UTC