kfac-jax

kfac-jax

google-deepmind

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

286 Stars
28 Forks
286 Watchers
Python Language
apache-2.0 License
Cost to Build
$35.4K
Market Value
$106.4K

Growth over time

4 data points  ·  2022-08-01 → 2025-09-01
Stars Forks Watchers
💬

How do you feel about this project?

Ask AI about kfac-jax

Question copied to clipboard

What is the google-deepmind/kfac-jax GitHub project? Description: "Second Order Optimization and Curvature Estimation with K-FAC in JAX.". Written in Python. Explain what it does, its main use cases, key features, and who would benefit from using it.

Question is copied to clipboard — paste it after the AI opens.

How to clone kfac-jax

Clone via HTTPS

git clone https://github.com/google-deepmind/kfac-jax.git

Clone via SSH

[email protected]:google-deepmind/kfac-jax.git

Download ZIP

Download master.zip

Found an issue?

Report bugs or request features on the kfac-jax issue tracker:

Open GitHub Issues