google-deepmind / optax
Optax is a gradient processing and optimization library for JAX.
- Repository stars
- 2,365All-time #23,723
- Fork
- 391
- Last push
- updated 4 days ago
- Repository created
- since 2020
About
Optax is a gradient processing and optimization library for JAX.Summarised by AI from the README; the original prevails
Author description: Optax is a gradient processing and optimization library for JAX.
Basic listing based on official public information; functionality has not been independently evaluated.
- Pricing
- Pricing not verifiedCheck current official software, hosting, model and infrastructure pricing.