google/jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX. observed · 2026-08-28
Health v2 · maintenance only
72/100
- Activity 85
- Release rhythm 40
- Longevity 100
How is this computed?
round(0.45*activity + 0.35*rhythm + 0.20*longevity); archived -> min(score, 10) — computed 2026-09-03. Adoption (stars, forks) is never an input.
- gap_med: 4
- age_days: 1878
- days_rel: 506
- days_push: 90
- n_releases_24m: 2
Adoption not part of the score
1059 stars · 76 forks observed · 2026-08-28
What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded
JAXopt is a Python library providing hardware-accelerated, batchable, and differentiable optimizers built on JAX, including implicit differentiation of optimization problem solutions. The project is no longer maintained, with many features ported into Optax.
Use cases
- differentiate through optimization problems in JAX
- run optimizers on GPU and TPU
- vectorize many instances of an optimization problem with vmap
- implicit differentiation of fixed points and constrained problems
- bilevel optimization and hyperparameter tuning
- root finding and solvers in JAX pipelines
When to choose
- you need differentiable or bilevel optimization in JAX and are fine using an unmaintained library
- you want implicit differentiation of optimization solutions
- you need batched solvers accelerated on GPU/TPU
When to avoid
- starting a new project - use Optax or other JAX alternatives instead
- you need maintained software with ongoing support
- you need optimizers outside the JAX ecosystem
Facets
library · maturity abandoned
machine-learning math machine-learning deep-learning python jax optimization implicit-differentiation bilevel-optimization root-finding differentiable-programming algorithms gpu
3 sources
- readme: https://github.com/google/jaxopt · fetched 2026-08-28 · 2b9bdd3865d7
- homepage: https://jaxopt.github.io · fetched 2026-08-29 · 9b0494b66f1b
- registry_pypi: https://pypi.org/pypi/jaxopt/json · fetched 2026-08-29 · 54203dfc1dba
Member repositories
| Repository | Role | Health v2 |
|---|---|---|
| google/jaxopt | main | 72 |
For agents
Data as of 2026-08-30T08:39:29.467469+00:00 · Report a problem