piwheels

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

Dependencies

• distrax
• absl-py
• jaxlib
• numpy
• dm-tree
• immutabledict
• optax
• typing-extensions
• jax

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
0.0.8	2026-02-25
kfac_jax-0.0.8-py3-none-any.whl (164 KB)
0.0.7	2025-05-20
kfac_jax-0.0.7-py3-none-any.whl (157 KB)
0.0.6	2024-04-04
kfac_jax-0.0.6-py3-none-any.whl (143 KB)
0.0.5	2023-05-16
kfac_jax-0.0.5-py3-none-any.whl (136 KB)
0.0.3	2022-09-23
kfac_jax-0.0.3-py3-none-any.whl (123 KB)
0.0.2	2022-06-07
kfac_jax-0.0.2-py3-none-any.whl (115 KB)
0.0.1	2022-04-01
kfac_jax-0.0.1-py3-none-any.whl (115 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue