Sealbook / Open source / Machine learning & data science

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

machine-learningoptimization

GitHub ↗Website ↗

License & maintenance

Apache-2.0 is a permissive license: free to use commercially with attribution.

Last code push 10 days ago.

Similar projects

	Repository	Stars	Language	License	Activity
1	tensorflow/tensorflowAn Open Source Machine Learning Framework for Everyone	★ 201.1k	C++	Apache-2.0	Active
2	pytorch/pytorchTensors and Dynamic neural networks in Python with strong GPU acceleration	★ 104.8k	Python	Other	Active
3	d2l-ai/d2l-zh《动手学深度学习》：面向中文读者、能运行、可讨论。中英文版被70多个国家的500多所大学用于教学。	★ 81.9k	Python	Apache-2.0	Stale
4	tesseract-ocr/tesseractTesseract Open Source OCR Engine (main repository)	★ 77.1k	C++	Apache-2.0	Active
5	openbq-org/OpenBBOpen Data Platform for analysts, quants and AI agents.	★ 74.4k	Python	Other	Active
6	microsoft/AI-For-Beginners12 Weeks, 24 Lessons, AI for All!	★ 69.9k	Jupyter Notebook	MIT	Active
7	scikit-learn/scikit-learnscikit-learn: machine learning in Python	★ 67.7k	Python	BSD-3-Clause	Active
8	labmlai/annotated_deep_learning_paper_implementations🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback,	★ 67.6k	Python	MIT	Slowing