Skip to content
This repository was archived by the owner on Apr 19, 2026. It is now read-only.
tensorflowPublic archive

About

An implementation of KFAC for TensorFlow

Resources

Stars

198 stars

Watchers

14 watching

Forks

Latest commit

 

History

127 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

K-FAC: Kronecker-Factored Approximate Curvature

Travis

K-FAC in TensorFlow is an implementation of K-FAC, an approximate second-order optimization method, in TensorFlow.

Installation

kfac is compatible with Python 2 and 3 and can be installed directly via pip,

# Assumes tensorflow or tensorflow-gpu installed
$ pip install kfac

# Installs with tensorflow-gpu requirement
$ pip install 'kfac[tensorflow_gpu]'

# Installs with tensorflow (cpu) requirement
$ pip install 'kfac[tensorflow]'

KFAC DOCS

Please check KFAC docs for a detailed description with examples of how to use KFAC. Check the Keras KFAC docs for information on using KFAC with Keras.

About

An implementation of KFAC for TensorFlow

Resources

Stars

198 stars

Watchers

14 watching

Forks

Releases

Packages

Contributors

Languages