optax

google-deepmind

Optax is a gradient processing and optimization library for JAX.

machine-learning optimization

View on GitHub Website

2.2k Stars

324 Forks

2.2k Watchers

Python Language

apache-2.0 License

100 SrcLog Score

Cost to Build

$541.5K

Market Value

$2.83M

How is this calculated?

Growth over time

3 data points · 2025-04-01 → 2026-04-01

Stars Forks Watchers

💬

How do you feel about this project?

Ask AI about optax

Question copied to clipboard

What is the google-deepmind/optax GitHub project? Description: "Optax is a gradient processing and optimization library for JAX.". Written in Python. Explain what it does, its main use cases, key features, and who would benefit from using it.

Question is copied to clipboard — paste it after the AI opens.

How to clone optax

Clone via HTTPS

git clone https://github.com/google-deepmind/optax.git

Clone via SSH

[email protected]:google-deepmind/optax.git

Download ZIP

Download master.zip

Found an issue?

Report bugs or request features on the optax issue tracker:

Open GitHub Issues

Similar to optax

tensorflow keras scikit-learn caffe TensorFlow-Examples tesseract machine-learning-for-software-engineers awesome-deep-learning-papers face_recognition CNTK incubator-mxnet data-science-ipython-notebooks lectures cs-video-courses julia Qix Screenshot-to-code spaCy cheatsheets-ai awesome-deep-learning python-machine-learning-book WaveFunctionCollapse handson-ml ML-From-Scratch gun dive-into-machine-learning tfjs tflearn awesome-datascience openpose