{"adoption": {"forks": 592, "observed_at": "2026-08-28T04:06:44.211945+00:00", "stars": 2408}, "canonical_url": "https://ross.abutalabs.com/products/maxtext", "card": {"archived": false, "artifact_type": "library", "description": "A simple, performant, and scalable Jax LLM!", "domain": ["large-language-models", "deep-learning", "machine-learning"], "enriched": true, "function": ["llm-training", "machine-learning", "deep-learning"], "health_score": 100, "homepage": "https://maxtext.readthedocs.io", "language": "Python", "license": "Apache-2.0", "license_family": "permissive", "maturity": "active", "member_repos": ["AI-Hypercomputer/maxtext"], "name": "AI-Hypercomputer/maxtext", "platform": ["python", "cloud"], "pushed_at": "2026-08-26T21:47:23+00:00", "repo": "AI-Hypercomputer/maxtext", "stars": 2408, "tags": ["jax", "tpu", "fine-tuning", "pre-training", "sft", "grpo", "gemma", "llama", "deepseek", "mistral", "xla", "gpu"], "topics": ["large-language-models", "llm", "gpt", "deepseek", "fine-tuning", "gemma2", "gemma3", "jax", "llama2", "llama3", "llama4", "mistral", "mixtral", "sft"], "urls": [], "use_cases": ["pre-train large language models on TPU clusters", "fine-tune Llama or Gemma with supervised fine-tuning", "run reinforcement learning post-training like GRPO for LLMs", "scale LLM training to tens of thousands of chips", "train DeepSeek or Mistral models in JAX", "benchmark high MFU LLM training on GPUs and TPUs"], "what_it_is": "MaxText is a high-performance, scalable open-source LLM training library written in pure Python/JAX, targeting Google Cloud TPUs and GPUs. It provides reference implementations of models like Gemma, Llama, DeepSeek, Qwen, and Mistral with support for pre-training at massive scale and post-training techniques such as SFT, GRPO, and GSPO.", "when_to_avoid": ["you need PyTorch-based training or ecosystem compatibility", "you only need LLM inference rather than training", "you lack access to TPU or GPU hardware", "you need diffusion or image generation models (see MaxDiffusion instead)"], "when_to_choose": ["you need to train or fine-tune LLMs on Google Cloud TPUs or GPUs at large scale", "you want a pure Python/JAX codebase that is easy to fork and modify", "you need high hardware utilization without hand-written kernel optimizations", "you want reference implementations of popular open model architectures"]}, "data_as_of": "2026-08-30T08:39:29.467469+00:00", "members": [{"path": "/products/maxtext", "repo": "AI-Hypercomputer/maxtext", "role": "main", "score": 97}], "provenance": {"archived": {"kind": "observed", "observed_at": "2026-08-28T04:06:44.211945+00:00", "source": "github"}, "artifact_type": {"confidence": null, "enriched_at": "2026-08-30T02:34:02.334993+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "120db8c0fb0e72e6714424fda3bf17604845b843a7f903a5d0bc613b45e8c615", "fetched_at": "2026-08-28T04:06:44.211945+00:00", "kind": "readme", "missing": false, "url": "https://github.com/AI-Hypercomputer/maxtext"}, {"content_hash": "4e4a2656cf1472b1735c7496eae37a5859c6864e253f2d2f3facde2e78997e87", "fetched_at": "2026-08-29T10:14:41.709723+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/maxtext/json"}], "taxonomy_version": 1}, "description": {"kind": "observed", "observed_at": "2026-08-28T04:06:44.211945+00:00", "source": "github"}, "domain": {"confidence": null, "enriched_at": "2026-08-30T02:34:02.334993+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "120db8c0fb0e72e6714424fda3bf17604845b843a7f903a5d0bc613b45e8c615", "fetched_at": "2026-08-28T04:06:44.211945+00:00", "kind": "readme", "missing": false, "url": "https://github.com/AI-Hypercomputer/maxtext"}, {"content_hash": "4e4a2656cf1472b1735c7496eae37a5859c6864e253f2d2f3facde2e78997e87", "fetched_at": "2026-08-29T10:14:41.709723+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/maxtext/json"}], "taxonomy_version": 1}, "enriched": {"inputs": [], "kind": "computed", "method": "enrichment_status"}, "function": {"confidence": null, "enriched_at": "2026-08-30T02:34:02.334993+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "120db8c0fb0e72e6714424fda3bf17604845b843a7f903a5d0bc613b45e8c615", "fetched_at": "2026-08-28T04:06:44.211945+00:00", "kind": "readme", "missing": false, "url": "https://github.com/AI-Hypercomputer/maxtext"}, {"content_hash": "4e4a2656cf1472b1735c7496eae37a5859c6864e253f2d2f3facde2e78997e87", "fetched_at": "2026-08-29T10:14:41.709723+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/maxtext/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:44.211945+00:00", "source": "github"}, "language": {"kind": "observed", "observed_at": "2026-08-28T04:06:44.211945+00:00", "source": "github"}, "license": {"kind": "observed", "observed_at": "2026-08-28T04:06:44.211945+00:00", "source": "github"}, "license_family": {"inputs": ["license"], "kind": "computed", "method": "license_family"}, "maturity": {"confidence": null, "enriched_at": "2026-08-30T02:34:02.334993+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "120db8c0fb0e72e6714424fda3bf17604845b843a7f903a5d0bc613b45e8c615", "fetched_at": "2026-08-28T04:06:44.211945+00:00", "kind": "readme", "missing": false, "url": "https://github.com/AI-Hypercomputer/maxtext"}, {"content_hash": "4e4a2656cf1472b1735c7496eae37a5859c6864e253f2d2f3facde2e78997e87", "fetched_at": "2026-08-29T10:14:41.709723+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/maxtext/json"}], "taxonomy_version": 1}, "member_repos": {"kind": "observed", "observed_at": "2026-08-28T04:06:44.211945+00:00", "source": "github"}, "name": {"kind": "observed", "observed_at": "2026-08-28T04:06:44.211945+00:00", "source": "github"}, "platform": {"confidence": null, "enriched_at": "2026-08-30T02:34:02.334993+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "120db8c0fb0e72e6714424fda3bf17604845b843a7f903a5d0bc613b45e8c615", "fetched_at": "2026-08-28T04:06:44.211945+00:00", "kind": "readme", "missing": false, "url": "https://github.com/AI-Hypercomputer/maxtext"}, {"content_hash": "4e4a2656cf1472b1735c7496eae37a5859c6864e253f2d2f3facde2e78997e87", "fetched_at": "2026-08-29T10:14:41.709723+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/maxtext/json"}], "taxonomy_version": 1}, "pushed_at": {"kind": "observed", "observed_at": "2026-08-28T04:06:44.211945+00:00", "source": "github"}, "repo": {"kind": "observed", "observed_at": "2026-08-28T04:06:44.211945+00:00", "source": "github"}, "stars": {"kind": "observed", "observed_at": "2026-08-28T04:06:44.211945+00:00", "source": "github"}, "tags": {"confidence": null, "enriched_at": "2026-08-30T02:34:02.334993+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "120db8c0fb0e72e6714424fda3bf17604845b843a7f903a5d0bc613b45e8c615", "fetched_at": "2026-08-28T04:06:44.211945+00:00", "kind": "readme", "missing": false, "url": "https://github.com/AI-Hypercomputer/maxtext"}, {"content_hash": "4e4a2656cf1472b1735c7496eae37a5859c6864e253f2d2f3facde2e78997e87", "fetched_at": "2026-08-29T10:14:41.709723+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/maxtext/json"}], "taxonomy_version": 1}, "topics": {"kind": "observed", "observed_at": "2026-08-28T04:06:44.211945+00:00", "source": "github"}, "urls": {"kind": "observed", "observed_at": "2026-08-28T04:06:44.211945+00:00", "source": "github"}, "use_cases": {"confidence": null, "enriched_at": "2026-08-30T02:34:02.334993+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "120db8c0fb0e72e6714424fda3bf17604845b843a7f903a5d0bc613b45e8c615", "fetched_at": "2026-08-28T04:06:44.211945+00:00", "kind": "readme", "missing": false, "url": "https://github.com/AI-Hypercomputer/maxtext"}, {"content_hash": "4e4a2656cf1472b1735c7496eae37a5859c6864e253f2d2f3facde2e78997e87", "fetched_at": "2026-08-29T10:14:41.709723+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/maxtext/json"}], "taxonomy_version": 1}, "what_it_is": {"confidence": null, "enriched_at": "2026-08-30T02:34:02.334993+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "120db8c0fb0e72e6714424fda3bf17604845b843a7f903a5d0bc613b45e8c615", "fetched_at": "2026-08-28T04:06:44.211945+00:00", "kind": "readme", "missing": false, "url": "https://github.com/AI-Hypercomputer/maxtext"}, {"content_hash": "4e4a2656cf1472b1735c7496eae37a5859c6864e253f2d2f3facde2e78997e87", "fetched_at": "2026-08-29T10:14:41.709723+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/maxtext/json"}], "taxonomy_version": 1}, "when_to_avoid": {"confidence": null, "enriched_at": "2026-08-30T02:34:02.334993+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "120db8c0fb0e72e6714424fda3bf17604845b843a7f903a5d0bc613b45e8c615", "fetched_at": "2026-08-28T04:06:44.211945+00:00", "kind": "readme", "missing": false, "url": "https://github.com/AI-Hypercomputer/maxtext"}, {"content_hash": "4e4a2656cf1472b1735c7496eae37a5859c6864e253f2d2f3facde2e78997e87", "fetched_at": "2026-08-29T10:14:41.709723+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/maxtext/json"}], "taxonomy_version": 1}, "when_to_choose": {"confidence": null, "enriched_at": "2026-08-30T02:34:02.334993+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "120db8c0fb0e72e6714424fda3bf17604845b843a7f903a5d0bc613b45e8c615", "fetched_at": "2026-08-28T04:06:44.211945+00:00", "kind": "readme", "missing": false, "url": "https://github.com/AI-Hypercomputer/maxtext"}, {"content_hash": "4e4a2656cf1472b1735c7496eae37a5859c6864e253f2d2f3facde2e78997e87", "fetched_at": "2026-08-29T10:14:41.709723+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/maxtext/json"}], "taxonomy_version": 1}}, "score": {"components": {"activity": 99, "longevity": 91, "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": 1282, "days_push": 7, "days_rel": 12, "gap_med": 18, "n_releases_24m": 20}, "score": 97, "version": 2}, "staleness": {"enrichment_outdated": false, "low_confidence": false, "scrape_days": 9, "stale_scrape": false}}