{"adoption": {"forks": 830, "observed_at": "2026-08-28T04:09:58.132098+00:00", "stars": 7303}, "canonical_url": "https://ross.abutalabs.com/products/flax", "card": {"archived": false, "artifact_type": "library", "description": "Flax is a neural network library for JAX that is designed for flexibility.", "domain": ["machine-learning", "deep-learning", "artificial-intelligence"], "enriched": true, "function": ["machine-learning", "deep-learning", "llm-training"], "health_score": 100, "homepage": "https://flax.readthedocs.io", "language": "Jupyter Notebook", "license": "Apache-2.0", "license_family": "permissive", "maturity": "active", "member_repos": ["google/flax"], "name": "google/flax", "platform": ["python", "cross-platform"], "pushed_at": "2026-08-24T19:34:32+00:00", "repo": "google/flax", "stars": 7303, "tags": ["jax", "neural-networks", "nnx", "linen", "research", "google", "gpu"], "topics": ["jax"], "urls": [], "use_cases": ["build neural networks in jax", "train deep learning models with jax", "define models with regular python objects and mutable state", "run deep learning research with flexible training loops", "migrate from flax linen to the nnx api", "train models on gpus and tpus"], "what_it_is": "Flax is a neural network library and ecosystem for JAX designed for flexibility, featuring the newer NNX API with first-class Python reference semantics alongside the older Linen API. It is developed in close collaboration with the JAX team for high-performance deep learning research.", "when_to_avoid": ["you prefer PyTorch or TensorFlow-style frameworks", "you want a batteries-included high-level training framework with opinionated defaults", "your project depends on the legacy Linen API without plans to migrate"], "when_to_choose": ["you use JAX and want a flexible, research-friendly neural network library", "you want Python reference semantics with mutable, shareable model state (NNX)", "you need a mature, Google-backed ecosystem with examples and tutorials"]}, "data_as_of": "2026-08-30T08:39:29.467469+00:00", "members": [{"path": "/products/flax", "repo": "google/flax", "role": "main", "score": 99}], "provenance": {"archived": {"kind": "observed", "observed_at": "2026-08-28T04:09:58.132098+00:00", "source": "github"}, "artifact_type": {"confidence": null, "enriched_at": "2026-08-29T17:38:41.740331+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "9c7d9f7f40c477d20e7550043f58778ace9bd58cf518e58390c4d3ef573ab5a7", "fetched_at": "2026-08-28T04:09:58.132098+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google/flax"}, {"content_hash": "94ff4bfef308b98289010b032e91c95008ded27883b5bf444b207846660e28d7", "fetched_at": "2026-08-29T08:34:33.506554+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/flax/json"}], "taxonomy_version": 1}, "description": {"kind": "observed", "observed_at": "2026-08-28T04:09:58.132098+00:00", "source": "github"}, "domain": {"confidence": null, "enriched_at": "2026-08-29T17:38:41.740331+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "9c7d9f7f40c477d20e7550043f58778ace9bd58cf518e58390c4d3ef573ab5a7", "fetched_at": "2026-08-28T04:09:58.132098+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google/flax"}, {"content_hash": "94ff4bfef308b98289010b032e91c95008ded27883b5bf444b207846660e28d7", "fetched_at": "2026-08-29T08:34:33.506554+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/flax/json"}], "taxonomy_version": 1}, "enriched": {"inputs": [], "kind": "computed", "method": "enrichment_status"}, "function": {"confidence": null, "enriched_at": "2026-08-29T17:38:41.740331+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "9c7d9f7f40c477d20e7550043f58778ace9bd58cf518e58390c4d3ef573ab5a7", "fetched_at": "2026-08-28T04:09:58.132098+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google/flax"}, {"content_hash": "94ff4bfef308b98289010b032e91c95008ded27883b5bf444b207846660e28d7", "fetched_at": "2026-08-29T08:34:33.506554+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/flax/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:09:58.132098+00:00", "source": "github"}, "language": {"kind": "observed", "observed_at": "2026-08-28T04:09:58.132098+00:00", "source": "github"}, "license": {"kind": "observed", "observed_at": "2026-08-28T04:09:58.132098+00:00", "source": "github"}, "license_family": {"inputs": ["license"], "kind": "computed", "method": "license_family"}, "maturity": {"confidence": null, "enriched_at": "2026-08-29T17:38:41.740331+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "9c7d9f7f40c477d20e7550043f58778ace9bd58cf518e58390c4d3ef573ab5a7", "fetched_at": "2026-08-28T04:09:58.132098+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google/flax"}, {"content_hash": "94ff4bfef308b98289010b032e91c95008ded27883b5bf444b207846660e28d7", "fetched_at": "2026-08-29T08:34:33.506554+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/flax/json"}], "taxonomy_version": 1}, "member_repos": {"kind": "observed", "observed_at": "2026-08-28T04:09:58.132098+00:00", "source": "github"}, "name": {"kind": "observed", "observed_at": "2026-08-28T04:09:58.132098+00:00", "source": "github"}, "platform": {"confidence": null, "enriched_at": "2026-08-29T17:38:41.740331+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "9c7d9f7f40c477d20e7550043f58778ace9bd58cf518e58390c4d3ef573ab5a7", "fetched_at": "2026-08-28T04:09:58.132098+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google/flax"}, {"content_hash": "94ff4bfef308b98289010b032e91c95008ded27883b5bf444b207846660e28d7", "fetched_at": "2026-08-29T08:34:33.506554+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/flax/json"}], "taxonomy_version": 1}, "pushed_at": {"kind": "observed", "observed_at": "2026-08-28T04:09:58.132098+00:00", "source": "github"}, "repo": {"kind": "observed", "observed_at": "2026-08-28T04:09:58.132098+00:00", "source": "github"}, "stars": {"kind": "observed", "observed_at": "2026-08-28T04:09:58.132098+00:00", "source": "github"}, "tags": {"confidence": null, "enriched_at": "2026-08-29T17:38:41.740331+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "9c7d9f7f40c477d20e7550043f58778ace9bd58cf518e58390c4d3ef573ab5a7", "fetched_at": "2026-08-28T04:09:58.132098+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google/flax"}, {"content_hash": "94ff4bfef308b98289010b032e91c95008ded27883b5bf444b207846660e28d7", "fetched_at": "2026-08-29T08:34:33.506554+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/flax/json"}], "taxonomy_version": 1}, "topics": {"kind": "observed", "observed_at": "2026-08-28T04:09:58.132098+00:00", "source": "github"}, "urls": {"kind": "observed", "observed_at": "2026-08-28T04:09:58.132098+00:00", "source": "github"}, "use_cases": {"confidence": null, "enriched_at": "2026-08-29T17:38:41.740331+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "9c7d9f7f40c477d20e7550043f58778ace9bd58cf518e58390c4d3ef573ab5a7", "fetched_at": "2026-08-28T04:09:58.132098+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google/flax"}, {"content_hash": "94ff4bfef308b98289010b032e91c95008ded27883b5bf444b207846660e28d7", "fetched_at": "2026-08-29T08:34:33.506554+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/flax/json"}], "taxonomy_version": 1}, "what_it_is": {"confidence": null, "enriched_at": "2026-08-29T17:38:41.740331+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "9c7d9f7f40c477d20e7550043f58778ace9bd58cf518e58390c4d3ef573ab5a7", "fetched_at": "2026-08-28T04:09:58.132098+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google/flax"}, {"content_hash": "94ff4bfef308b98289010b032e91c95008ded27883b5bf444b207846660e28d7", "fetched_at": "2026-08-29T08:34:33.506554+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/flax/json"}], "taxonomy_version": 1}, "when_to_avoid": {"confidence": null, "enriched_at": "2026-08-29T17:38:41.740331+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "9c7d9f7f40c477d20e7550043f58778ace9bd58cf518e58390c4d3ef573ab5a7", "fetched_at": "2026-08-28T04:09:58.132098+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google/flax"}, {"content_hash": "94ff4bfef308b98289010b032e91c95008ded27883b5bf444b207846660e28d7", "fetched_at": "2026-08-29T08:34:33.506554+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/flax/json"}], "taxonomy_version": 1}, "when_to_choose": {"confidence": null, "enriched_at": "2026-08-29T17:38:41.740331+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "9c7d9f7f40c477d20e7550043f58778ace9bd58cf518e58390c4d3ef573ab5a7", "fetched_at": "2026-08-28T04:09:58.132098+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google/flax"}, {"content_hash": "94ff4bfef308b98289010b032e91c95008ded27883b5bf444b207846660e28d7", "fetched_at": "2026-08-29T08:34:33.506554+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/flax/json"}], "taxonomy_version": 1}}, "score": {"components": {"activity": 99, "longevity": 100, "rhythm": 98}, "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": 2427, "days_push": 9, "days_rel": 15, "gap_med": 27.5, "n_releases_24m": 21}, "score": 99, "version": 2}, "staleness": {"enrichment_outdated": false, "low_confidence": false, "scrape_days": 9, "stale_scrape": false}}