Sign in
← Back to search
github Actively maintained

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

1 awesome list

Quick read

Stars
2,324
Forks
361
Open issues
109
Commits
2,025

Activity and growth

Latest capture 2026-08-22 03:08

Stars · last 7 days
No history
Commits · last 7 days
No history
Stars since tracking
+55
Stored snapshots
8

Classification

Topics

#machine-learning #optimization

Metadata

Language
Python
License
Apache-2.0
Default branch
main
Created
2020-06-12
First commit
2020-07-09
Last pushed
2026-08-20
GitHub updated
2026-08-21
Last synced
2026-08-22 03:08
Stack scanned
2026-08-22 03:08
Archived
No
Sign in to like google-deepmind/optax
README Open website View on GitHub

AI development signals

0 paths

Agent instructions and tool configuration found in this repository.

No config files detected.

Growth history

Tracked growth

8 observed captures since 2026-05-22. Observed captures are shown by default.

Stars from first capture +55

Chart data

Observed captures only

Time horizon

All tracked data

Custom date range

Stars history

Observed snapshots

Commits history

Observed snapshots

Similar repositories

Nearest indexed repositories by embedding similarity.

patrick-kidger/equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

2,936 stars
Python 1 awesome list

jax-ml/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

36,252 stars
Python 3 awesome lists

facebook/Ax

Adaptive Experimentation Platform

2,790 stars
Python 1 awesome list

patrick-kidger/diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

2,078 stars
Python 1 awesome list