Training & fine-tuning
Optax
Builds optimizers from small gradient-processing components that can be recombined for custom algorithms.
Overview
Research summary
Optax supplies the optimization components used to update model parameters in JAX training programs. It includes common optimizers, loss functions and lower-level gradient transformations that developers can combine into customized update rules. Optimizer state and parameter updates remain explicit, making it suitable for both standard neural-network training and optimization research. The project complements libraries such as Flax rather than replacing model definitions or data loading. Its modular design makes individual optimization ideas reusable within a larger training loop.
Repository summary
- Stars
- 2,345
- Open issues
- Unavailable
- Last push
- 2026-10-01
- Commits, 90 days
- Unavailable
- Repository activity
- Not scored
- Version
- Unavailable
Recorded catalogue figures. View repository data and provenance →
Classification
Pricing & services
Paid services unknown
Whether the provider offers paid products or services has not been established.
Licence scope
Implementation
Recorded implementation details and interfaces for Optax.
Implementation details
Python library and supporting tools
- Languages
- Python
- Repository type
- source
Recorded interfaces and capabilities
Licence scope
Repository
Repository snapshots, release information and recorded maintenance signals.
Repository snapshot
- Stars
- 2,345
- Open issues
- Unavailable
- Last push
- 2026-10-01
- Commits, 90 days
- Unavailable
- Repository activity
- Not scored
- Archived
- Not recorded
Repository activity is a snapshot, not a quality or popularity ranking. It combines recent-push freshness (50%), 90-day commits (30%) and issue pressure (20%).
Maintenance and provenance
- Catalogue snapshot
- 2026-10-06
- Stars source
- Recorded fallback
- Last push source
- Recorded fallback
Documentation
Recorded references and research provenance for this entry.
Recorded sources 5
- https://optax.readthedocs.io Project page · Documentation
- https://github.com/google-deepmind/optax Linked repository · Research reference
- https://api.github.com/repos/google-deepmind/optax Research reference
- https://github.com/google-deepmind/optax/blob/main/README.md Research reference
- https://github.com/google-deepmind/optax/blob/main/LICENSE Research reference
Research metadata
- Research date
- 2026-10-02
- Catalogue snapshot
- 2026-10-06
Community & social 1 channels
Communities
- GitHub Discussionsgithub.com/google-deepmind/optax/discussionsOfficial
Link details for GitHub Discussions
- Last checked