{"adoption": {"forks": 877, "observed_at": "2026-08-28T04:09:42.952710+00:00", "stars": 6380}, "canonical_url": "https://ross.abutalabs.com/products/mesh-transformer-jax", "card": {"archived": false, "artifact_type": "library", "description": "Model parallel transformers in JAX and Haiku", "domain": ["large-language-models", "deep-learning", "machine-learning", "gpu-computing"], "enriched": true, "function": ["llm-training", "llm-inference", "deep-learning", "machine-learning"], "health_score": 20, "homepage": null, "language": "Python", "license": "Apache-2.0", "license_family": "permissive", "maturity": "maintenance", "member_repos": ["kingoflolz/mesh-transformer-jax"], "name": "kingoflolz/mesh-transformer-jax", "platform": ["python", "cloud"], "pushed_at": "2023-01-21T00:09:29+00:00", "repo": "kingoflolz/mesh-transformer-jax", "stars": 6380, "tags": ["jax", "haiku", "model-parallelism", "tpu", "gpt-j", "transformers", "megatron-style-parallelism", "gpu"], "topics": [], "urls": [], "use_cases": ["train large transformer models on TPUs with model parallelism", "run inference with GPT-J-6B", "fine-tune GPT-J-6B on custom datasets", "generate text with a 6B parameter open-source language model", "experiment with Megatron-style tensor parallelism in JAX"], "what_it_is": "A JAX/Haiku library implementing model-parallel training and inference of transformer models using xmap/pjit operators, similar to Megatron-LM's parallelism scheme, optimized for TPUs. It is the codebase behind GPT-J-6B, a 6 billion parameter autoregressive language model trained on The Pile.", "when_to_avoid": ["you need to scale beyond ~40B parameters (use GPT-NeoX or DeepSpeed instead)", "you are working on GPUs rather than TPUs", "you need a actively developed or general-purpose LLM training framework"], "when_to_choose": ["you need to train or fine-tune transformers up to ~40B parameters on TPU hardware", "you want to run or fine-tune GPT-J-6B", "you prefer JAX/Haiku over PyTorch for large-scale model parallelism"]}, "data_as_of": "2026-08-30T08:39:29.467469+00:00", "members": [{"path": "/products/mesh-transformer-jax", "repo": "kingoflolz/mesh-transformer-jax", "role": "main", "score": 32}], "provenance": {"archived": {"kind": "observed", "observed_at": "2026-08-28T04:09:42.952710+00:00", "source": "github"}, "artifact_type": {"confidence": null, "enriched_at": "2026-08-29T17:45:09.540793+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "6fc2f66d18e064366fa95fe39ba4cd2c148ac324f070b5c29a07c0d462cf2988", "fetched_at": "2026-08-28T04:09:42.952710+00:00", "kind": "readme", "missing": false, "url": "https://github.com/kingoflolz/mesh-transformer-jax"}], "taxonomy_version": 1}, "description": {"kind": "observed", "observed_at": "2026-08-28T04:09:42.952710+00:00", "source": "github"}, "domain": {"confidence": null, "enriched_at": "2026-08-29T17:45:09.540793+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "6fc2f66d18e064366fa95fe39ba4cd2c148ac324f070b5c29a07c0d462cf2988", "fetched_at": "2026-08-28T04:09:42.952710+00:00", "kind": "readme", "missing": false, "url": "https://github.com/kingoflolz/mesh-transformer-jax"}], "taxonomy_version": 1}, "enriched": {"inputs": [], "kind": "computed", "method": "enrichment_status"}, "function": {"confidence": null, "enriched_at": "2026-08-29T17:45:09.540793+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "6fc2f66d18e064366fa95fe39ba4cd2c148ac324f070b5c29a07c0d462cf2988", "fetched_at": "2026-08-28T04:09:42.952710+00:00", "kind": "readme", "missing": false, "url": "https://github.com/kingoflolz/mesh-transformer-jax"}], "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:42.952710+00:00", "source": "github"}, "language": {"kind": "observed", "observed_at": "2026-08-28T04:09:42.952710+00:00", "source": "github"}, "license": {"kind": "observed", "observed_at": "2026-08-28T04:09:42.952710+00:00", "source": "github"}, "license_family": {"inputs": ["license"], "kind": "computed", "method": "license_family"}, "maturity": {"confidence": null, "enriched_at": "2026-08-29T17:45:09.540793+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "6fc2f66d18e064366fa95fe39ba4cd2c148ac324f070b5c29a07c0d462cf2988", "fetched_at": "2026-08-28T04:09:42.952710+00:00", "kind": "readme", "missing": false, "url": "https://github.com/kingoflolz/mesh-transformer-jax"}], "taxonomy_version": 1}, "member_repos": {"kind": "observed", "observed_at": "2026-08-28T04:09:42.952710+00:00", "source": "github"}, "name": {"kind": "observed", "observed_at": "2026-08-28T04:09:42.952710+00:00", "source": "github"}, "platform": {"confidence": null, "enriched_at": "2026-08-29T17:45:09.540793+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "6fc2f66d18e064366fa95fe39ba4cd2c148ac324f070b5c29a07c0d462cf2988", "fetched_at": "2026-08-28T04:09:42.952710+00:00", "kind": "readme", "missing": false, "url": "https://github.com/kingoflolz/mesh-transformer-jax"}], "taxonomy_version": 1}, "pushed_at": {"kind": "observed", "observed_at": "2026-08-28T04:09:42.952710+00:00", "source": "github"}, "repo": {"kind": "observed", "observed_at": "2026-08-28T04:09:42.952710+00:00", "source": "github"}, "stars": {"kind": "observed", "observed_at": "2026-08-28T04:09:42.952710+00:00", "source": "github"}, "tags": {"confidence": null, "enriched_at": "2026-08-29T17:45:09.540793+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "6fc2f66d18e064366fa95fe39ba4cd2c148ac324f070b5c29a07c0d462cf2988", "fetched_at": "2026-08-28T04:09:42.952710+00:00", "kind": "readme", "missing": false, "url": "https://github.com/kingoflolz/mesh-transformer-jax"}], "taxonomy_version": 1}, "topics": {"kind": "observed", "observed_at": "2026-08-28T04:09:42.952710+00:00", "source": "github"}, "urls": {"kind": "observed", "observed_at": "2026-08-28T04:09:42.952710+00:00", "source": "github"}, "use_cases": {"confidence": null, "enriched_at": "2026-08-29T17:45:09.540793+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "6fc2f66d18e064366fa95fe39ba4cd2c148ac324f070b5c29a07c0d462cf2988", "fetched_at": "2026-08-28T04:09:42.952710+00:00", "kind": "readme", "missing": false, "url": "https://github.com/kingoflolz/mesh-transformer-jax"}], "taxonomy_version": 1}, "what_it_is": {"confidence": null, "enriched_at": "2026-08-29T17:45:09.540793+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "6fc2f66d18e064366fa95fe39ba4cd2c148ac324f070b5c29a07c0d462cf2988", "fetched_at": "2026-08-28T04:09:42.952710+00:00", "kind": "readme", "missing": false, "url": "https://github.com/kingoflolz/mesh-transformer-jax"}], "taxonomy_version": 1}, "when_to_avoid": {"confidence": null, "enriched_at": "2026-08-29T17:45:09.540793+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "6fc2f66d18e064366fa95fe39ba4cd2c148ac324f070b5c29a07c0d462cf2988", "fetched_at": "2026-08-28T04:09:42.952710+00:00", "kind": "readme", "missing": false, "url": "https://github.com/kingoflolz/mesh-transformer-jax"}], "taxonomy_version": 1}, "when_to_choose": {"confidence": null, "enriched_at": "2026-08-29T17:45:09.540793+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "6fc2f66d18e064366fa95fe39ba4cd2c148ac324f070b5c29a07c0d462cf2988", "fetched_at": "2026-08-28T04:09:42.952710+00:00", "kind": "readme", "missing": false, "url": "https://github.com/kingoflolz/mesh-transformer-jax"}], "taxonomy_version": 1}}, "score": {"components": {"activity": 0, "longevity": 100, "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": 1999, "days_push": 1321, "days_rel": null, "gap_med": null, "n_releases_24m": 0}, "score": 32, "version": 2}, "staleness": {"enrichment_outdated": false, "low_confidence": false, "scrape_days": 9, "stale_scrape": false}}