228 Forks
1851 Stars
1851 Watchers

optax

Optax is a gradient processing and optimization library for JAX.

How to download and setup optax

Open terminal and run command
git clone https://github.com/google-deepmind/optax.git
git clone is used to create a copy or clone of optax 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 optax https://github.com/google-deepmind/optax/archive/master.zip

Or simply clone optax with SSH
[email protected]:google-deepmind/optax.git

If you have some problems with optax

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