{"adoption": {"forks": 406, "observed_at": "2026-08-28T04:06:41.859669+00:00", "stars": 2372}, "canonical_url": "https://ross.abutalabs.com/products/axlearn", "card": {"archived": false, "artifact_type": "library", "description": "An Extensible Deep Learning Library", "domain": ["deep-learning", "machine-learning", "large-language-models", "computer-vision", "speech-processing", "gpu-computing"], "enriched": true, "function": ["deep-learning", "llm-training", "machine-learning", "cli"], "health_score": 75, "homepage": null, "language": "Python", "license": "Apache-2.0", "license_family": "permissive", "maturity": "active", "member_repos": ["apple/axlearn"], "name": "apple/axlearn", "platform": ["python", "cloud"], "pushed_at": "2026-07-08T21:26:38+00:00", "repo": "apple/axlearn", "stars": 2372, "tags": ["jax", "xla", "gspmd", "large-scale-training", "distributed-training", "model-training", "flax", "huggingface", "natural-language-processing", "gpu", "linux"], "topics": ["deep-learning", "jax"], "urls": [], "use_cases": ["train large language models on thousands of accelerators", "build deep learning models with a composable configuration system", "run large-scale training jobs on public clouds", "train speech recognition or computer vision models at scale", "integrate Flax or Hugging Face transformers models into a JAX training pipeline", "manage and deploy distributed training jobs and data"], "what_it_is": "AXLearn is a Python deep learning library built on JAX and XLA for developing and training large-scale models, with an object-oriented configuration system for composing reusable model components. It scales to hundreds of billions of parameters across thousands of accelerators and supports NLP, vision, and speech workloads on public clouds.", "when_to_avoid": ["you need a stable API, since it is under active development and subject to change", "you want a lightweight library for small models or quick experiments", "you prefer PyTorch or TensorFlow ecosystems", "you lack access to large accelerator clusters"], "when_to_choose": ["you need to train models with tens or hundreds of billions of parameters", "you want JAX/XLA-based training with high accelerator utilization", "you need cloud tooling for deploying and managing large training jobs", "you prefer an object-oriented, modular approach to model composition"]}, "data_as_of": "2026-08-30T08:39:29.467469+00:00", "members": [{"path": "/products/axlearn", "repo": "apple/axlearn", "role": "main", "score": 71}], "provenance": {"archived": {"kind": "observed", "observed_at": "2026-08-28T04:06:41.859669+00:00", "source": "github"}, "artifact_type": {"confidence": null, "enriched_at": "2026-08-30T02:35:09.026887+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "98a0349e319770033b3ce3a842c3e07c23295d843b25a489ce6cb8c89750fac4", "fetched_at": "2026-08-28T04:06:41.859669+00:00", "kind": "readme", "missing": false, "url": "https://github.com/apple/axlearn"}], "taxonomy_version": 1}, "description": {"kind": "observed", "observed_at": "2026-08-28T04:06:41.859669+00:00", "source": "github"}, "domain": {"confidence": null, "enriched_at": "2026-08-30T02:35:09.026887+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "98a0349e319770033b3ce3a842c3e07c23295d843b25a489ce6cb8c89750fac4", "fetched_at": "2026-08-28T04:06:41.859669+00:00", "kind": "readme", "missing": false, "url": "https://github.com/apple/axlearn"}], "taxonomy_version": 1}, "enriched": {"inputs": [], "kind": "computed", "method": "enrichment_status"}, "function": {"confidence": null, "enriched_at": "2026-08-30T02:35:09.026887+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "98a0349e319770033b3ce3a842c3e07c23295d843b25a489ce6cb8c89750fac4", "fetched_at": "2026-08-28T04:06:41.859669+00:00", "kind": "readme", "missing": false, "url": "https://github.com/apple/axlearn"}], "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:41.859669+00:00", "source": "github"}, "language": {"kind": "observed", "observed_at": "2026-08-28T04:06:41.859669+00:00", "source": "github"}, "license": {"kind": "observed", "observed_at": "2026-08-28T04:06:41.859669+00:00", "source": "github"}, "license_family": {"inputs": ["license"], "kind": "computed", "method": "license_family"}, "maturity": {"confidence": null, "enriched_at": "2026-08-30T02:35:09.026887+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "98a0349e319770033b3ce3a842c3e07c23295d843b25a489ce6cb8c89750fac4", "fetched_at": "2026-08-28T04:06:41.859669+00:00", "kind": "readme", "missing": false, "url": "https://github.com/apple/axlearn"}], "taxonomy_version": 1}, "member_repos": {"kind": "observed", "observed_at": "2026-08-28T04:06:41.859669+00:00", "source": "github"}, "name": {"kind": "observed", "observed_at": "2026-08-28T04:06:41.859669+00:00", "source": "github"}, "platform": {"confidence": null, "enriched_at": "2026-08-30T02:35:09.026887+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "98a0349e319770033b3ce3a842c3e07c23295d843b25a489ce6cb8c89750fac4", "fetched_at": "2026-08-28T04:06:41.859669+00:00", "kind": "readme", "missing": false, "url": "https://github.com/apple/axlearn"}], "taxonomy_version": 1}, "pushed_at": {"kind": "observed", "observed_at": "2026-08-28T04:06:41.859669+00:00", "source": "github"}, "repo": {"kind": "observed", "observed_at": "2026-08-28T04:06:41.859669+00:00", "source": "github"}, "stars": {"kind": "observed", "observed_at": "2026-08-28T04:06:41.859669+00:00", "source": "github"}, "tags": {"confidence": null, "enriched_at": "2026-08-30T02:35:09.026887+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "98a0349e319770033b3ce3a842c3e07c23295d843b25a489ce6cb8c89750fac4", "fetched_at": "2026-08-28T04:06:41.859669+00:00", "kind": "readme", "missing": false, "url": "https://github.com/apple/axlearn"}], "taxonomy_version": 1}, "topics": {"kind": "observed", "observed_at": "2026-08-28T04:06:41.859669+00:00", "source": "github"}, "urls": {"kind": "observed", "observed_at": "2026-08-28T04:06:41.859669+00:00", "source": "github"}, "use_cases": {"confidence": null, "enriched_at": "2026-08-30T02:35:09.026887+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "98a0349e319770033b3ce3a842c3e07c23295d843b25a489ce6cb8c89750fac4", "fetched_at": "2026-08-28T04:06:41.859669+00:00", "kind": "readme", "missing": false, "url": "https://github.com/apple/axlearn"}], "taxonomy_version": 1}, "what_it_is": {"confidence": null, "enriched_at": "2026-08-30T02:35:09.026887+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "98a0349e319770033b3ce3a842c3e07c23295d843b25a489ce6cb8c89750fac4", "fetched_at": "2026-08-28T04:06:41.859669+00:00", "kind": "readme", "missing": false, "url": "https://github.com/apple/axlearn"}], "taxonomy_version": 1}, "when_to_avoid": {"confidence": null, "enriched_at": "2026-08-30T02:35:09.026887+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "98a0349e319770033b3ce3a842c3e07c23295d843b25a489ce6cb8c89750fac4", "fetched_at": "2026-08-28T04:06:41.859669+00:00", "kind": "readme", "missing": false, "url": "https://github.com/apple/axlearn"}], "taxonomy_version": 1}, "when_to_choose": {"confidence": null, "enriched_at": "2026-08-30T02:35:09.026887+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "98a0349e319770033b3ce3a842c3e07c23295d843b25a489ce6cb8c89750fac4", "fetched_at": "2026-08-28T04:06:41.859669+00:00", "kind": "readme", "missing": false, "url": "https://github.com/apple/axlearn"}], "taxonomy_version": 1}}, "score": {"components": {"activity": 91, "longevity": 91, "rhythm": 35}, "computed_at": "2026-09-03T02:20:16.233290+00:00", "flags": ["no_releases"], "formula": "round(0.45*activity + 0.35*rhythm + 0.20*longevity); archived -> min(score, 10)", "inputs": {"age_days": 1286, "days_push": 56, "days_rel": null, "gap_med": null, "n_releases_24m": 0}, "score": 71, "version": 2}, "staleness": {"enrichment_outdated": false, "low_confidence": false, "scrape_days": 9, "stale_scrape": false}}