Repository · GitHub

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

Owner	google-deepmind
License (as declared)	Apache-2.0
Stars	2,350
Forks	379
Main language	Python
Created	12 June 2020
Last updated at source	9 October 2026

pythonmachine-learningoptimization

Embed this entry

<iframe src="https://silytics.si/embed/item/f3b7cd9d-f4ee-4217-a0ea-cea61b5ba914" width="420" height="190" style="border:0" loading="lazy" title="Silytics entry"></iframe>

Free, no tracking. More about feeds and embeds.

More from google-deepmind

sonnetRepository
pysc2Repository
labRepository
open_spielRepository
graph_netsRepository
dm_controlRepository

Silytics shows public metadata from GitHub and links to the original. Information may be incomplete or out of date. Last refreshed 15 October 2026. Licenses and terms of the original apply to the work itself. Not financial, legal or professional advice.