{"adoption": {"forks": 3744, "observed_at": "2026-08-28T04:12:00.999641+00:00", "stars": 36215}, "canonical_url": "https://ross.abutalabs.com/products/jax", "card": {"archived": false, "artifact_type": "library", "description": "Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more", "domain": ["machine-learning", "deep-learning", "gpu-computing", "data-science"], "enriched": true, "function": ["machine-learning", "deep-learning", "math", "compiler", "gpu-computing", "llm-training"], "health_score": 100, "homepage": "https://docs.jax.dev", "language": "Python", "license": "Apache-2.0", "license_family": "permissive", "maturity": "stable", "member_repos": ["jax-ml/jax"], "name": "jax-ml/jax", "platform": ["python", "windows", "cloud"], "pushed_at": "2026-08-27T00:18:16+00:00", "repo": "jax-ml/jax", "stars": 36215, "tags": ["autodiff", "jit-compilation", "xla", "numpy", "tpu", "vectorization", "pallas", "numerical-computing", "algorithms", "linux", "macos", "gpu"], "topics": ["jax"], "urls": [], "use_cases": ["train neural networks with automatic differentiation in Python", "JIT-compile NumPy-style code to run on GPUs and TPUs", "vectorize per-example gradient computations with vmap", "write custom GPU/TPU kernels with Pallas", "run large-scale machine learning research across accelerators", "compute higher-order derivatives like Jacobians and Hessians", "parallelize and shard computations across multiple devices"], "what_it_is": "JAX is a Python library for accelerator-oriented array computation and composable program transformations, offering NumPy-compatible APIs with automatic differentiation, vectorization, and JIT compilation via XLA. It scales the same code across CPU, GPU, and TPU backends and includes Pallas for writing custom hardware kernels.", "when_to_avoid": ["you need a batteries-included deep learning framework with prebuilt layers and training loops out of the box", "your project depends on impure Python code with side effects inside compiled functions", "you need stable experimental APIs like Pallas Triton or unsupported hardware backends", "you only need simple CPU array math where plain NumPy suffices"], "when_to_choose": ["you need high-performance numerical computing or ML research on GPU/TPU", "you want composable autodiff, batching, and compilation in pure Python", "you prefer a NumPy-like API with accelerator scaling", "you need custom low-level kernels for TPU or Hopper+ GPUs"]}, "data_as_of": "2026-08-30T08:39:29.467469+00:00", "members": [{"path": "/products/jax", "repo": "jax-ml/jax", "role": "main", "score": 99}], "provenance": {"archived": {"kind": "observed", "observed_at": "2026-08-28T04:12:00.999641+00:00", "source": "github"}, "artifact_type": {"confidence": null, "enriched_at": "2026-08-29T16:26:43.094812+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "7c101dab4b0fd06ee7858c7dd7d899fb3f80785e07b0839d696a198ae8bc4d20", "fetched_at": "2026-08-28T04:12:00.999641+00:00", "kind": "readme", "missing": false, "url": "https://github.com/jax-ml/jax"}, {"content_hash": "7a112cd10fa70de3204b11a3a408465df484ff0aa93a3fff8acf8db974976aca", "fetched_at": "2026-08-29T07:47:46.993762+00:00", "kind": "homepage", "missing": false, "url": "https://docs.jax.dev"}, {"content_hash": "392db64cd25d7e8e4fc1d0aa04576f1e7a62bd3282a077361c71299e3d591600", "fetched_at": "2026-08-29T07:47:47.018800+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/quickstart.html"}, {"content_hash": "1ac6612ed34fbf01568cb1252df6423dcd052e7d6a457e65f1d7bc7a483f5585", "fetched_at": "2026-08-29T07:47:47.021460+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/CHANGELOG.html"}, {"content_hash": "dce08330eb76f3744748042329b33cdb52b0892724438d00f0aeb700cf15bc02", "fetched_at": "2026-08-29T07:47:47.024717+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/about.html"}, {"content_hash": "c4cb7ecaba1805b10929157bf32a409e2293f74307c449fc4fca566d9cbee30d", "fetched_at": "2026-08-29T07:47:47.026494+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/faq.html"}, {"content_hash": "271ba08ebebd90a568e860b950d77cd3ad665fe4a6b8060a2b5d813e3b25111b", "fetched_at": "2026-08-29T07:47:47.012178+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/installation.html"}, {"content_hash": "76c783c3a03721da5d77fc0abb04597cabf2b025d42129747715f05b45c1321d", "fetched_at": "2026-08-29T07:47:47.014732+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/tpu/quickstart.html"}, {"content_hash": "bdc9e6b9b200597c306d155422754fdbb19d7a1d53a6b76173734ec35193ac8c", "fetched_at": "2026-08-29T07:47:47.016791+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/gpu/quickstart.html"}, {"content_hash": "b7dcbe7aa32ce6171a7e0b92bda39ed7ec35d3e19267d583eae461844a700d3e", "fetched_at": "2026-08-29T07:47:47.028550+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/changelog.html"}], "taxonomy_version": 1}, "description": {"kind": "observed", "observed_at": "2026-08-28T04:12:00.999641+00:00", "source": "github"}, "domain": {"confidence": null, "enriched_at": "2026-08-29T16:26:43.094812+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "7c101dab4b0fd06ee7858c7dd7d899fb3f80785e07b0839d696a198ae8bc4d20", "fetched_at": "2026-08-28T04:12:00.999641+00:00", "kind": "readme", "missing": false, "url": "https://github.com/jax-ml/jax"}, {"content_hash": "7a112cd10fa70de3204b11a3a408465df484ff0aa93a3fff8acf8db974976aca", "fetched_at": "2026-08-29T07:47:46.993762+00:00", "kind": "homepage", "missing": false, "url": "https://docs.jax.dev"}, {"content_hash": "392db64cd25d7e8e4fc1d0aa04576f1e7a62bd3282a077361c71299e3d591600", "fetched_at": "2026-08-29T07:47:47.018800+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/quickstart.html"}, {"content_hash": "1ac6612ed34fbf01568cb1252df6423dcd052e7d6a457e65f1d7bc7a483f5585", "fetched_at": "2026-08-29T07:47:47.021460+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/CHANGELOG.html"}, {"content_hash": "dce08330eb76f3744748042329b33cdb52b0892724438d00f0aeb700cf15bc02", "fetched_at": "2026-08-29T07:47:47.024717+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/about.html"}, {"content_hash": "c4cb7ecaba1805b10929157bf32a409e2293f74307c449fc4fca566d9cbee30d", "fetched_at": "2026-08-29T07:47:47.026494+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/faq.html"}, {"content_hash": "271ba08ebebd90a568e860b950d77cd3ad665fe4a6b8060a2b5d813e3b25111b", "fetched_at": "2026-08-29T07:47:47.012178+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/installation.html"}, {"content_hash": "76c783c3a03721da5d77fc0abb04597cabf2b025d42129747715f05b45c1321d", "fetched_at": "2026-08-29T07:47:47.014732+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/tpu/quickstart.html"}, {"content_hash": "bdc9e6b9b200597c306d155422754fdbb19d7a1d53a6b76173734ec35193ac8c", "fetched_at": "2026-08-29T07:47:47.016791+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/gpu/quickstart.html"}, {"content_hash": "b7dcbe7aa32ce6171a7e0b92bda39ed7ec35d3e19267d583eae461844a700d3e", "fetched_at": "2026-08-29T07:47:47.028550+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/changelog.html"}], "taxonomy_version": 1}, "enriched": {"inputs": [], "kind": "computed", "method": "enrichment_status"}, "function": {"confidence": null, "enriched_at": "2026-08-29T16:26:43.094812+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "7c101dab4b0fd06ee7858c7dd7d899fb3f80785e07b0839d696a198ae8bc4d20", "fetched_at": "2026-08-28T04:12:00.999641+00:00", "kind": "readme", "missing": false, "url": "https://github.com/jax-ml/jax"}, {"content_hash": "7a112cd10fa70de3204b11a3a408465df484ff0aa93a3fff8acf8db974976aca", "fetched_at": "2026-08-29T07:47:46.993762+00:00", "kind": "homepage", "missing": false, "url": "https://docs.jax.dev"}, {"content_hash": "392db64cd25d7e8e4fc1d0aa04576f1e7a62bd3282a077361c71299e3d591600", "fetched_at": "2026-08-29T07:47:47.018800+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/quickstart.html"}, {"content_hash": "1ac6612ed34fbf01568cb1252df6423dcd052e7d6a457e65f1d7bc7a483f5585", "fetched_at": "2026-08-29T07:47:47.021460+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/CHANGELOG.html"}, {"content_hash": "dce08330eb76f3744748042329b33cdb52b0892724438d00f0aeb700cf15bc02", "fetched_at": "2026-08-29T07:47:47.024717+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/about.html"}, {"content_hash": "c4cb7ecaba1805b10929157bf32a409e2293f74307c449fc4fca566d9cbee30d", "fetched_at": "2026-08-29T07:47:47.026494+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/faq.html"}, {"content_hash": "271ba08ebebd90a568e860b950d77cd3ad665fe4a6b8060a2b5d813e3b25111b", "fetched_at": "2026-08-29T07:47:47.012178+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/installation.html"}, {"content_hash": "76c783c3a03721da5d77fc0abb04597cabf2b025d42129747715f05b45c1321d", "fetched_at": "2026-08-29T07:47:47.014732+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/tpu/quickstart.html"}, {"content_hash": "bdc9e6b9b200597c306d155422754fdbb19d7a1d53a6b76173734ec35193ac8c", "fetched_at": "2026-08-29T07:47:47.016791+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/gpu/quickstart.html"}, {"content_hash": "b7dcbe7aa32ce6171a7e0b92bda39ed7ec35d3e19267d583eae461844a700d3e", "fetched_at": "2026-08-29T07:47:47.028550+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/changelog.html"}], "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:12:00.999641+00:00", "source": "github"}, "language": {"kind": "observed", "observed_at": "2026-08-28T04:12:00.999641+00:00", "source": "github"}, "license": {"kind": "observed", "observed_at": "2026-08-28T04:12:00.999641+00:00", "source": "github"}, "license_family": {"inputs": ["license"], "kind": "computed", "method": "license_family"}, "maturity": {"confidence": null, "enriched_at": "2026-08-29T16:26:43.094812+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "7c101dab4b0fd06ee7858c7dd7d899fb3f80785e07b0839d696a198ae8bc4d20", "fetched_at": "2026-08-28T04:12:00.999641+00:00", "kind": "readme", "missing": false, "url": "https://github.com/jax-ml/jax"}, {"content_hash": "7a112cd10fa70de3204b11a3a408465df484ff0aa93a3fff8acf8db974976aca", "fetched_at": "2026-08-29T07:47:46.993762+00:00", "kind": "homepage", "missing": false, "url": "https://docs.jax.dev"}, {"content_hash": "392db64cd25d7e8e4fc1d0aa04576f1e7a62bd3282a077361c71299e3d591600", "fetched_at": "2026-08-29T07:47:47.018800+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/quickstart.html"}, {"content_hash": "1ac6612ed34fbf01568cb1252df6423dcd052e7d6a457e65f1d7bc7a483f5585", "fetched_at": "2026-08-29T07:47:47.021460+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/CHANGELOG.html"}, {"content_hash": "dce08330eb76f3744748042329b33cdb52b0892724438d00f0aeb700cf15bc02", "fetched_at": "2026-08-29T07:47:47.024717+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/about.html"}, {"content_hash": "c4cb7ecaba1805b10929157bf32a409e2293f74307c449fc4fca566d9cbee30d", "fetched_at": "2026-08-29T07:47:47.026494+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/faq.html"}, {"content_hash": "271ba08ebebd90a568e860b950d77cd3ad665fe4a6b8060a2b5d813e3b25111b", "fetched_at": "2026-08-29T07:47:47.012178+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/installation.html"}, {"content_hash": "76c783c3a03721da5d77fc0abb04597cabf2b025d42129747715f05b45c1321d", "fetched_at": "2026-08-29T07:47:47.014732+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/tpu/quickstart.html"}, {"content_hash": "bdc9e6b9b200597c306d155422754fdbb19d7a1d53a6b76173734ec35193ac8c", "fetched_at": "2026-08-29T07:47:47.016791+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/gpu/quickstart.html"}, {"content_hash": "b7dcbe7aa32ce6171a7e0b92bda39ed7ec35d3e19267d583eae461844a700d3e", "fetched_at": "2026-08-29T07:47:47.028550+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/changelog.html"}], "taxonomy_version": 1}, "member_repos": {"kind": "observed", "observed_at": "2026-08-28T04:12:00.999641+00:00", "source": "github"}, "name": {"kind": "observed", "observed_at": "2026-08-28T04:12:00.999641+00:00", "source": "github"}, "platform": {"confidence": null, "enriched_at": "2026-08-29T16:26:43.094812+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "7c101dab4b0fd06ee7858c7dd7d899fb3f80785e07b0839d696a198ae8bc4d20", "fetched_at": "2026-08-28T04:12:00.999641+00:00", "kind": "readme", "missing": false, "url": "https://github.com/jax-ml/jax"}, {"content_hash": "7a112cd10fa70de3204b11a3a408465df484ff0aa93a3fff8acf8db974976aca", "fetched_at": "2026-08-29T07:47:46.993762+00:00", "kind": "homepage", "missing": false, "url": "https://docs.jax.dev"}, {"content_hash": "392db64cd25d7e8e4fc1d0aa04576f1e7a62bd3282a077361c71299e3d591600", "fetched_at": "2026-08-29T07:47:47.018800+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/quickstart.html"}, {"content_hash": "1ac6612ed34fbf01568cb1252df6423dcd052e7d6a457e65f1d7bc7a483f5585", "fetched_at": "2026-08-29T07:47:47.021460+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/CHANGELOG.html"}, {"content_hash": "dce08330eb76f3744748042329b33cdb52b0892724438d00f0aeb700cf15bc02", "fetched_at": "2026-08-29T07:47:47.024717+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/about.html"}, {"content_hash": "c4cb7ecaba1805b10929157bf32a409e2293f74307c449fc4fca566d9cbee30d", "fetched_at": "2026-08-29T07:47:47.026494+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/faq.html"}, {"content_hash": "271ba08ebebd90a568e860b950d77cd3ad665fe4a6b8060a2b5d813e3b25111b", "fetched_at": "2026-08-29T07:47:47.012178+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/installation.html"}, {"content_hash": "76c783c3a03721da5d77fc0abb04597cabf2b025d42129747715f05b45c1321d", "fetched_at": "2026-08-29T07:47:47.014732+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/tpu/quickstart.html"}, {"content_hash": "bdc9e6b9b200597c306d155422754fdbb19d7a1d53a6b76173734ec35193ac8c", "fetched_at": "2026-08-29T07:47:47.016791+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/gpu/quickstart.html"}, {"content_hash": "b7dcbe7aa32ce6171a7e0b92bda39ed7ec35d3e19267d583eae461844a700d3e", "fetched_at": "2026-08-29T07:47:47.028550+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/changelog.html"}], "taxonomy_version": 1}, "pushed_at": {"kind": "observed", "observed_at": "2026-08-28T04:12:00.999641+00:00", "source": "github"}, "repo": {"kind": "observed", "observed_at": "2026-08-28T04:12:00.999641+00:00", "source": "github"}, "stars": {"kind": "observed", "observed_at": "2026-08-28T04:12:00.999641+00:00", "source": "github"}, "tags": {"confidence": null, "enriched_at": "2026-08-29T16:26:43.094812+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "7c101dab4b0fd06ee7858c7dd7d899fb3f80785e07b0839d696a198ae8bc4d20", "fetched_at": "2026-08-28T04:12:00.999641+00:00", "kind": "readme", "missing": false, "url": "https://github.com/jax-ml/jax"}, {"content_hash": "7a112cd10fa70de3204b11a3a408465df484ff0aa93a3fff8acf8db974976aca", "fetched_at": "2026-08-29T07:47:46.993762+00:00", "kind": "homepage", "missing": false, "url": "https://docs.jax.dev"}, {"content_hash": "392db64cd25d7e8e4fc1d0aa04576f1e7a62bd3282a077361c71299e3d591600", "fetched_at": "2026-08-29T07:47:47.018800+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/quickstart.html"}, {"content_hash": "1ac6612ed34fbf01568cb1252df6423dcd052e7d6a457e65f1d7bc7a483f5585", "fetched_at": "2026-08-29T07:47:47.021460+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/CHANGELOG.html"}, {"content_hash": "dce08330eb76f3744748042329b33cdb52b0892724438d00f0aeb700cf15bc02", "fetched_at": "2026-08-29T07:47:47.024717+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/about.html"}, {"content_hash": "c4cb7ecaba1805b10929157bf32a409e2293f74307c449fc4fca566d9cbee30d", "fetched_at": "2026-08-29T07:47:47.026494+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/faq.html"}, {"content_hash": "271ba08ebebd90a568e860b950d77cd3ad665fe4a6b8060a2b5d813e3b25111b", "fetched_at": "2026-08-29T07:47:47.012178+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/installation.html"}, {"content_hash": "76c783c3a03721da5d77fc0abb04597cabf2b025d42129747715f05b45c1321d", "fetched_at": "2026-08-29T07:47:47.014732+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/tpu/quickstart.html"}, {"content_hash": "bdc9e6b9b200597c306d155422754fdbb19d7a1d53a6b76173734ec35193ac8c", "fetched_at": "2026-08-29T07:47:47.016791+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/gpu/quickstart.html"}, {"content_hash": "b7dcbe7aa32ce6171a7e0b92bda39ed7ec35d3e19267d583eae461844a700d3e", "fetched_at": "2026-08-29T07:47:47.028550+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/changelog.html"}], "taxonomy_version": 1}, "topics": {"kind": "observed", "observed_at": "2026-08-28T04:12:00.999641+00:00", "source": "github"}, "urls": {"kind": "observed", "observed_at": "2026-08-28T04:12:00.999641+00:00", "source": "github"}, "use_cases": {"confidence": null, "enriched_at": "2026-08-29T16:26:43.094812+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "7c101dab4b0fd06ee7858c7dd7d899fb3f80785e07b0839d696a198ae8bc4d20", "fetched_at": "2026-08-28T04:12:00.999641+00:00", "kind": "readme", "missing": false, "url": "https://github.com/jax-ml/jax"}, {"content_hash": "7a112cd10fa70de3204b11a3a408465df484ff0aa93a3fff8acf8db974976aca", "fetched_at": "2026-08-29T07:47:46.993762+00:00", "kind": "homepage", "missing": false, "url": "https://docs.jax.dev"}, {"content_hash": "392db64cd25d7e8e4fc1d0aa04576f1e7a62bd3282a077361c71299e3d591600", "fetched_at": "2026-08-29T07:47:47.018800+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/quickstart.html"}, {"content_hash": "1ac6612ed34fbf01568cb1252df6423dcd052e7d6a457e65f1d7bc7a483f5585", "fetched_at": "2026-08-29T07:47:47.021460+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/CHANGELOG.html"}, {"content_hash": "dce08330eb76f3744748042329b33cdb52b0892724438d00f0aeb700cf15bc02", "fetched_at": "2026-08-29T07:47:47.024717+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/about.html"}, {"content_hash": "c4cb7ecaba1805b10929157bf32a409e2293f74307c449fc4fca566d9cbee30d", "fetched_at": "2026-08-29T07:47:47.026494+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/faq.html"}, {"content_hash": "271ba08ebebd90a568e860b950d77cd3ad665fe4a6b8060a2b5d813e3b25111b", "fetched_at": "2026-08-29T07:47:47.012178+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/installation.html"}, {"content_hash": "76c783c3a03721da5d77fc0abb04597cabf2b025d42129747715f05b45c1321d", "fetched_at": "2026-08-29T07:47:47.014732+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/tpu/quickstart.html"}, {"content_hash": "bdc9e6b9b200597c306d155422754fdbb19d7a1d53a6b76173734ec35193ac8c", "fetched_at": "2026-08-29T07:47:47.016791+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/gpu/quickstart.html"}, {"content_hash": "b7dcbe7aa32ce6171a7e0b92bda39ed7ec35d3e19267d583eae461844a700d3e", "fetched_at": "2026-08-29T07:47:47.028550+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/changelog.html"}], "taxonomy_version": 1}, "what_it_is": {"confidence": null, "enriched_at": "2026-08-29T16:26:43.094812+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "7c101dab4b0fd06ee7858c7dd7d899fb3f80785e07b0839d696a198ae8bc4d20", "fetched_at": "2026-08-28T04:12:00.999641+00:00", "kind": "readme", "missing": false, "url": "https://github.com/jax-ml/jax"}, {"content_hash": "7a112cd10fa70de3204b11a3a408465df484ff0aa93a3fff8acf8db974976aca", "fetched_at": "2026-08-29T07:47:46.993762+00:00", "kind": "homepage", "missing": false, "url": "https://docs.jax.dev"}, {"content_hash": "392db64cd25d7e8e4fc1d0aa04576f1e7a62bd3282a077361c71299e3d591600", "fetched_at": "2026-08-29T07:47:47.018800+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/quickstart.html"}, {"content_hash": "1ac6612ed34fbf01568cb1252df6423dcd052e7d6a457e65f1d7bc7a483f5585", "fetched_at": "2026-08-29T07:47:47.021460+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/CHANGELOG.html"}, {"content_hash": "dce08330eb76f3744748042329b33cdb52b0892724438d00f0aeb700cf15bc02", "fetched_at": "2026-08-29T07:47:47.024717+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/about.html"}, {"content_hash": "c4cb7ecaba1805b10929157bf32a409e2293f74307c449fc4fca566d9cbee30d", "fetched_at": "2026-08-29T07:47:47.026494+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/faq.html"}, {"content_hash": "271ba08ebebd90a568e860b950d77cd3ad665fe4a6b8060a2b5d813e3b25111b", "fetched_at": "2026-08-29T07:47:47.012178+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/installation.html"}, {"content_hash": "76c783c3a03721da5d77fc0abb04597cabf2b025d42129747715f05b45c1321d", "fetched_at": "2026-08-29T07:47:47.014732+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/tpu/quickstart.html"}, {"content_hash": "bdc9e6b9b200597c306d155422754fdbb19d7a1d53a6b76173734ec35193ac8c", "fetched_at": "2026-08-29T07:47:47.016791+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/gpu/quickstart.html"}, {"content_hash": "b7dcbe7aa32ce6171a7e0b92bda39ed7ec35d3e19267d583eae461844a700d3e", "fetched_at": "2026-08-29T07:47:47.028550+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/changelog.html"}], "taxonomy_version": 1}, "when_to_avoid": {"confidence": null, "enriched_at": "2026-08-29T16:26:43.094812+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "7c101dab4b0fd06ee7858c7dd7d899fb3f80785e07b0839d696a198ae8bc4d20", "fetched_at": "2026-08-28T04:12:00.999641+00:00", "kind": "readme", "missing": false, "url": "https://github.com/jax-ml/jax"}, {"content_hash": "7a112cd10fa70de3204b11a3a408465df484ff0aa93a3fff8acf8db974976aca", "fetched_at": "2026-08-29T07:47:46.993762+00:00", "kind": "homepage", "missing": false, "url": "https://docs.jax.dev"}, {"content_hash": "392db64cd25d7e8e4fc1d0aa04576f1e7a62bd3282a077361c71299e3d591600", "fetched_at": "2026-08-29T07:47:47.018800+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/quickstart.html"}, {"content_hash": "1ac6612ed34fbf01568cb1252df6423dcd052e7d6a457e65f1d7bc7a483f5585", "fetched_at": "2026-08-29T07:47:47.021460+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/CHANGELOG.html"}, {"content_hash": "dce08330eb76f3744748042329b33cdb52b0892724438d00f0aeb700cf15bc02", "fetched_at": "2026-08-29T07:47:47.024717+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/about.html"}, {"content_hash": "c4cb7ecaba1805b10929157bf32a409e2293f74307c449fc4fca566d9cbee30d", "fetched_at": "2026-08-29T07:47:47.026494+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/faq.html"}, {"content_hash": "271ba08ebebd90a568e860b950d77cd3ad665fe4a6b8060a2b5d813e3b25111b", "fetched_at": "2026-08-29T07:47:47.012178+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/installation.html"}, {"content_hash": "76c783c3a03721da5d77fc0abb04597cabf2b025d42129747715f05b45c1321d", "fetched_at": "2026-08-29T07:47:47.014732+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/tpu/quickstart.html"}, {"content_hash": "bdc9e6b9b200597c306d155422754fdbb19d7a1d53a6b76173734ec35193ac8c", "fetched_at": "2026-08-29T07:47:47.016791+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/gpu/quickstart.html"}, {"content_hash": "b7dcbe7aa32ce6171a7e0b92bda39ed7ec35d3e19267d583eae461844a700d3e", "fetched_at": "2026-08-29T07:47:47.028550+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/changelog.html"}], "taxonomy_version": 1}, "when_to_choose": {"confidence": null, "enriched_at": "2026-08-29T16:26:43.094812+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "7c101dab4b0fd06ee7858c7dd7d899fb3f80785e07b0839d696a198ae8bc4d20", "fetched_at": "2026-08-28T04:12:00.999641+00:00", "kind": "readme", "missing": false, "url": "https://github.com/jax-ml/jax"}, {"content_hash": "7a112cd10fa70de3204b11a3a408465df484ff0aa93a3fff8acf8db974976aca", "fetched_at": "2026-08-29T07:47:46.993762+00:00", "kind": "homepage", "missing": false, "url": "https://docs.jax.dev"}, {"content_hash": "392db64cd25d7e8e4fc1d0aa04576f1e7a62bd3282a077361c71299e3d591600", "fetched_at": "2026-08-29T07:47:47.018800+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/quickstart.html"}, {"content_hash": "1ac6612ed34fbf01568cb1252df6423dcd052e7d6a457e65f1d7bc7a483f5585", "fetched_at": "2026-08-29T07:47:47.021460+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/CHANGELOG.html"}, {"content_hash": "dce08330eb76f3744748042329b33cdb52b0892724438d00f0aeb700cf15bc02", "fetched_at": "2026-08-29T07:47:47.024717+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/about.html"}, {"content_hash": "c4cb7ecaba1805b10929157bf32a409e2293f74307c449fc4fca566d9cbee30d", "fetched_at": "2026-08-29T07:47:47.026494+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/faq.html"}, {"content_hash": "271ba08ebebd90a568e860b950d77cd3ad665fe4a6b8060a2b5d813e3b25111b", "fetched_at": "2026-08-29T07:47:47.012178+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/installation.html"}, {"content_hash": "76c783c3a03721da5d77fc0abb04597cabf2b025d42129747715f05b45c1321d", "fetched_at": "2026-08-29T07:47:47.014732+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/tpu/quickstart.html"}, {"content_hash": "bdc9e6b9b200597c306d155422754fdbb19d7a1d53a6b76173734ec35193ac8c", "fetched_at": "2026-08-29T07:47:47.016791+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/pallas/gpu/quickstart.html"}, {"content_hash": "b7dcbe7aa32ce6171a7e0b92bda39ed7ec35d3e19267d583eae461844a700d3e", "fetched_at": "2026-08-29T07:47:47.028550+00:00", "kind": "site_page", "missing": false, "url": "https://docs.jax.dev/en/latest/changelog.html"}], "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": 2869, "days_push": 7, "days_rel": 16, "gap_med": 28.0, "n_releases_24m": 31}, "score": 99, "version": 2}, "staleness": {"enrichment_outdated": false, "low_confidence": false, "scrape_days": 9, "stale_scrape": false}}