{"adoption": {"forks": 361, "observed_at": "2026-08-28T04:06:37.360091+00:00", "stars": 2325}, "canonical_url": "https://ross.abutalabs.com/products/optax", "card": {"archived": false, "artifact_type": "library", "description": "Optax is a gradient processing and optimization library for JAX.", "domain": ["machine-learning", "deep-learning", "reinforcement-learning"], "enriched": true, "function": ["machine-learning", "llm-training", "deep-learning"], "health_score": 97, "homepage": "https://optax.readthedocs.io", "language": "Python", "license": "Apache-2.0", "license_family": "permissive", "maturity": "stable", "member_repos": ["google-deepmind/optax"], "name": "google-deepmind/optax", "platform": ["python", "cross-platform"], "pushed_at": "2026-08-20T07:32:16+00:00", "repo": "google-deepmind/optax", "stars": 2325, "tags": ["jax", "optimizers", "gradient-processing", "loss-functions", "deepmind", "algorithms", "gpu"], "topics": ["machine-learning", "optimization"], "urls": [], "use_cases": ["train neural networks in jax with adam optimizer", "implement custom optimizers from composable gradient transforms", "compute loss functions for machine learning models", "run gradient descent research experiments in jax", "optimize reinforcement learning policies", "schedule learning rates during training"], "what_it_is": "Optax is a gradient processing and optimization library for JAX, offering composable building blocks like optimizers and loss functions. It is developed by DeepMind to support machine learning research with easily recombined components.", "when_to_avoid": ["you work in PyTorch or TensorFlow rather than JAX", "you need a full high-level training framework with data loading and experiment tracking"], "when_to_choose": ["you use JAX and need well-tested optimizers or loss functions", "you want to compose custom gradient processing pipelines", "you need readable implementations that match standard equations"]}, "data_as_of": "2026-08-30T08:39:29.467469+00:00", "members": [{"path": "/products/optax", "repo": "google-deepmind/optax", "role": "main", "score": 82}], "provenance": {"archived": {"kind": "observed", "observed_at": "2026-08-28T04:06:37.360091+00:00", "source": "github"}, "artifact_type": {"confidence": null, "enriched_at": "2026-08-30T02:38:24.040560+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "ceef840458cff00226d076dbbdcff551b24c8791bd3e24d1e69e095288be4994", "fetched_at": "2026-08-28T04:06:37.360091+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/optax"}, {"content_hash": "bf601364e27e7fc592d4ad089a160fff9d16454dceb4dfb5c6b7bce301d8d344", "fetched_at": "2026-08-29T10:19:12.351620+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/optax/json"}], "taxonomy_version": 1}, "description": {"kind": "observed", "observed_at": "2026-08-28T04:06:37.360091+00:00", "source": "github"}, "domain": {"confidence": null, "enriched_at": "2026-08-30T02:38:24.040560+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "ceef840458cff00226d076dbbdcff551b24c8791bd3e24d1e69e095288be4994", "fetched_at": "2026-08-28T04:06:37.360091+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/optax"}, {"content_hash": "bf601364e27e7fc592d4ad089a160fff9d16454dceb4dfb5c6b7bce301d8d344", "fetched_at": "2026-08-29T10:19:12.351620+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/optax/json"}], "taxonomy_version": 1}, "enriched": {"inputs": [], "kind": "computed", "method": "enrichment_status"}, "function": {"confidence": null, "enriched_at": "2026-08-30T02:38:24.040560+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "ceef840458cff00226d076dbbdcff551b24c8791bd3e24d1e69e095288be4994", "fetched_at": "2026-08-28T04:06:37.360091+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/optax"}, {"content_hash": "bf601364e27e7fc592d4ad089a160fff9d16454dceb4dfb5c6b7bce301d8d344", "fetched_at": "2026-08-29T10:19:12.351620+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/optax/json"}], "taxonomy_version": 1}, "health_score": {"inputs": ["days_since_push", "days_since_release", "archived"], "kind": "computed", "method": "health_v1"}, "homepage": {"kind": "observed", "observed_at": "2026-08-28T04:06:37.360091+00:00", "source": "github"}, "language": {"kind": "observed", "observed_at": "2026-08-28T04:06:37.360091+00:00", "source": "github"}, "license": {"kind": "observed", "observed_at": "2026-08-28T04:06:37.360091+00:00", "source": "github"}, "license_family": {"inputs": ["license"], "kind": "computed", "method": "license_family"}, "maturity": {"confidence": null, "enriched_at": "2026-08-30T02:38:24.040560+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "ceef840458cff00226d076dbbdcff551b24c8791bd3e24d1e69e095288be4994", "fetched_at": "2026-08-28T04:06:37.360091+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/optax"}, {"content_hash": "bf601364e27e7fc592d4ad089a160fff9d16454dceb4dfb5c6b7bce301d8d344", "fetched_at": "2026-08-29T10:19:12.351620+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/optax/json"}], "taxonomy_version": 1}, "member_repos": {"kind": "observed", "observed_at": "2026-08-28T04:06:37.360091+00:00", "source": "github"}, "name": {"kind": "observed", "observed_at": "2026-08-28T04:06:37.360091+00:00", "source": "github"}, "platform": {"confidence": null, "enriched_at": "2026-08-30T02:38:24.040560+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "ceef840458cff00226d076dbbdcff551b24c8791bd3e24d1e69e095288be4994", "fetched_at": "2026-08-28T04:06:37.360091+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/optax"}, {"content_hash": "bf601364e27e7fc592d4ad089a160fff9d16454dceb4dfb5c6b7bce301d8d344", "fetched_at": "2026-08-29T10:19:12.351620+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/optax/json"}], "taxonomy_version": 1}, "pushed_at": {"kind": "observed", "observed_at": "2026-08-28T04:06:37.360091+00:00", "source": "github"}, "repo": {"kind": "observed", "observed_at": "2026-08-28T04:06:37.360091+00:00", "source": "github"}, "stars": {"kind": "observed", "observed_at": "2026-08-28T04:06:37.360091+00:00", "source": "github"}, "tags": {"confidence": null, "enriched_at": "2026-08-30T02:38:24.040560+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "ceef840458cff00226d076dbbdcff551b24c8791bd3e24d1e69e095288be4994", "fetched_at": "2026-08-28T04:06:37.360091+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/optax"}, {"content_hash": "bf601364e27e7fc592d4ad089a160fff9d16454dceb4dfb5c6b7bce301d8d344", "fetched_at": "2026-08-29T10:19:12.351620+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/optax/json"}], "taxonomy_version": 1}, "topics": {"kind": "observed", "observed_at": "2026-08-28T04:06:37.360091+00:00", "source": "github"}, "urls": {"kind": "observed", "observed_at": "2026-08-28T04:06:37.360091+00:00", "source": "github"}, "use_cases": {"confidence": null, "enriched_at": "2026-08-30T02:38:24.040560+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "ceef840458cff00226d076dbbdcff551b24c8791bd3e24d1e69e095288be4994", "fetched_at": "2026-08-28T04:06:37.360091+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/optax"}, {"content_hash": "bf601364e27e7fc592d4ad089a160fff9d16454dceb4dfb5c6b7bce301d8d344", "fetched_at": "2026-08-29T10:19:12.351620+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/optax/json"}], "taxonomy_version": 1}, "what_it_is": {"confidence": null, "enriched_at": "2026-08-30T02:38:24.040560+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "ceef840458cff00226d076dbbdcff551b24c8791bd3e24d1e69e095288be4994", "fetched_at": "2026-08-28T04:06:37.360091+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/optax"}, {"content_hash": "bf601364e27e7fc592d4ad089a160fff9d16454dceb4dfb5c6b7bce301d8d344", "fetched_at": "2026-08-29T10:19:12.351620+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/optax/json"}], "taxonomy_version": 1}, "when_to_avoid": {"confidence": null, "enriched_at": "2026-08-30T02:38:24.040560+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "ceef840458cff00226d076dbbdcff551b24c8791bd3e24d1e69e095288be4994", "fetched_at": "2026-08-28T04:06:37.360091+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/optax"}, {"content_hash": "bf601364e27e7fc592d4ad089a160fff9d16454dceb4dfb5c6b7bce301d8d344", "fetched_at": "2026-08-29T10:19:12.351620+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/optax/json"}], "taxonomy_version": 1}, "when_to_choose": {"confidence": null, "enriched_at": "2026-08-30T02:38:24.040560+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "ceef840458cff00226d076dbbdcff551b24c8791bd3e24d1e69e095288be4994", "fetched_at": "2026-08-28T04:06:37.360091+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/optax"}, {"content_hash": "bf601364e27e7fc592d4ad089a160fff9d16454dceb4dfb5c6b7bce301d8d344", "fetched_at": "2026-08-29T10:19:12.351620+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/optax/json"}], "taxonomy_version": 1}}, "score": {"components": {"activity": 98, "longevity": 100, "rhythm": 51}, "computed_at": "2026-09-03T02:20:16.233290+00:00", "flags": [], "formula": "round(0.45*activity + 0.35*rhythm + 0.20*longevity); archived -> min(score, 10)", "inputs": {"age_days": 2273, "days_push": 13, "days_rel": 166, "gap_med": 119.5, "n_releases_24m": 5}, "score": 82, "version": 2}, "staleness": {"enrichment_outdated": false, "low_confidence": false, "scrape_days": 9, "stale_scrape": false}}