google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

Python2,269 stars327 forksApache-2.0

machine-learningoptimization

Rankings

Daily

#5880

Weekly

#10250

Monthly

#11441

Total stars

Daily new stars

Get badge

This repository is tracked by Trending Repos. The badge upgrades automatically if it ever cracks the top 100.

Markdown

![Trending Repos](https://trending-repos.com/badge/google-deepmind/optax.svg)

HTML

<img src="https://trending-repos.com/badge/google-deepmind/optax.svg" alt="Trending Repos" />

Raw URL

https://trending-repos.com/badge/google-deepmind/optax.svg