google-deepmind/optax
View on GitHubOptax is a gradient processing and optimization library for JAX.
- Stars
- 2.3k
- Forks
- 369
- Open beginner issues
- 0
- Indexed issues
- 31
- Avg merge
- 10h 15m
- Merged PRs (30d)
- 7
- Dominant language
- Python
- License
- Apache-2.0
- Last GitHub push
- Sep 13, 2026
- Latest indexed
- Sep 16, 2026
- Contributing guide
- Contributing guide
- Code of conduct
- No code of conduct
- Beginner labels
- help wanted
-
type:feature
google-deepmind/optax#24 · 4 comments · 0 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#353 · 5 comments · 1 reaction · 0 assignees ·
-
help wanted type:feature
google-deepmind/optax#357 · 10 comments · 2 reactions · 0 assignees ·
-
type:support
google-deepmind/optax#384 · 2 comments · 1 reaction · 0 assignees ·
-
type:bug
google-deepmind/optax#412 · 0 comments · 2 reactions · 1 assignee ·
-
type:feature
google-deepmind/optax#432 · 3 comments · 4 reactions · 0 assignees ·
-
type:bug
google-deepmind/optax#472 · 17 comments · 0 reactions · 1 assignee ·
-
documentation type:feature
google-deepmind/optax#652 · 2 comments · 0 reactions · 1 assignee ·
-
type:others
google-deepmind/optax#977 · 35 comments · 21 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#1053 · 4 comments · 3 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#1098 · 0 comments · 0 reactions · 1 assignee ·
-
type:feature
google-deepmind/optax#1214 · 8 comments · 0 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#1280 · 14 comments · 2 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#1342 · 4 comments · 0 reactions · 0 assignees ·
-
dependencies type:feature
google-deepmind/optax#1372 · 3 comments · 1 reaction · 0 assignees ·
-
type:feature
google-deepmind/optax#1388 · 3 comments · 0 reactions · 0 assignees ·
-
type:others
google-deepmind/optax#1509 · 6 comments · 1 reaction · 1 assignee ·
-
type:feature
google-deepmind/optax#1528 · 20 comments · 0 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#1550 · 3 comments · 0 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#1561 · 1 comment · 0 reactions · 0 assignees ·
-
Implementing AdaMuon Opentype:feature
google-deepmind/optax#1575 · 4 comments · 1 reaction · 0 assignees ·
-
type:feature
google-deepmind/optax#1580 · 5 comments · 0 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#1603 · 6 comments · 0 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#1621 · 0 comments · 0 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#1658 · 1 comment · 0 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#1660 · 1 comment · 2 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#1708 · 4 comments · 0 reactions · 0 assignees ·
-
Adam-family optimizers (adam, amsgrad, adabelief, yogi) produce NaN on float16 zero-gradient steps Opentype:bug
google-deepmind/optax#1754 · 0 comments · 0 reactions · 0 assignees ·
-
type:feature
google-deepmind/optax#1770 · 0 comments · 0 reactions · 0 assignees ·
-
type:bug
google-deepmind/optax#1771 · 0 comments · 0 reactions · 0 assignees ·
-
type:bug
google-deepmind/optax#1784 · 0 comments · 0 reactions · 0 assignees ·