8 Forks
121 Stars
121 Watchers

kfac-jax

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

How to download and setup kfac-jax

Open terminal and run command
git clone https://github.com/deepmind/kfac-jax.git
git clone is used to create a copy or clone of kfac-jax repositories. You pass git clone a repository URL.
it supports a few different network protocols and corresponding URL formats.

Also you may download zip file with kfac-jax https://github.com/deepmind/kfac-jax/archive/master.zip

Or simply clone kfac-jax with SSH
[email protected]:deepmind/kfac-jax.git

If you have some problems with kfac-jax

You may open issue on kfac-jax support forum (system) here: https://github.com/deepmind/kfac-jax/issues