{"adoption": {"forks": 106, "observed_at": "2026-08-28T04:04:49.717961+00:00", "stars": 1474}, "canonical_url": "https://ross.abutalabs.com/products/jraph", "card": {"archived": true, "artifact_type": "library", "description": "A Graph Neural Network Library in Jax", "domain": ["machine-learning", "deep-learning", "graph-processing"], "enriched": true, "function": ["machine-learning", "deep-learning"], "health_score": 10, "homepage": "https://jraph.readthedocs.io/en/latest/", "language": "Python", "license": "Apache-2.0", "license_family": "permissive", "maturity": "maintenance", "member_repos": ["google-deepmind/jraph"], "name": "google-deepmind/jraph", "platform": ["python", "cross-platform"], "pushed_at": "2024-03-18T13:56:39+00:00", "repo": "google-deepmind/jraph", "stars": 1474, "tags": ["graph-neural-networks", "jax", "message-passing", "graphs", "gpu"], "topics": ["graph-neural-networks", "jax", "machine-learning", "deep-learning"], "urls": [], "use_cases": ["train graph neural networks in jax", "batch and pad variable-sized graphs for jit compilation", "run message passing neural networks on molecules", "distribute large graph networks across multiple devices", "learn graph neural networks with tutorials"], "what_it_is": "Jraph is a lightweight Graph Neural Network library built on JAX, providing a GraphsTuple data structure, graph utilities, and a zoo of forkable GNN models. It integrates with Haiku or Flax for parameter management and supports distributed training of large graphs.", "when_to_avoid": ["you want a framework that manages model parameters for you", "you need PyTorch or TensorFlow GNN support", "you need actively developed features with frequent releases"], "when_to_choose": ["you use JAX and need GNN primitives you can fork and adapt", "you want lightweight graph data structures and utilities without a heavy framework", "you need to scale large MPNNs across devices with pmap"]}, "data_as_of": "2026-08-30T08:39:29.467469+00:00", "members": [{"path": "/products/jraph", "repo": "google-deepmind/jraph", "role": "main", "score": 10}], "provenance": {"archived": {"kind": "observed", "observed_at": "2026-08-28T04:04:49.717961+00:00", "source": "github"}, "artifact_type": {"confidence": null, "enriched_at": "2026-08-30T04:34:33.644239+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "c74ad85491a11353909917d1213202999599f0e3734009755ba6c0d3b87d546a", "fetched_at": "2026-08-28T04:04:49.717961+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/jraph"}], "taxonomy_version": 1}, "description": {"kind": "observed", "observed_at": "2026-08-28T04:04:49.717961+00:00", "source": "github"}, "domain": {"confidence": null, "enriched_at": "2026-08-30T04:34:33.644239+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "c74ad85491a11353909917d1213202999599f0e3734009755ba6c0d3b87d546a", "fetched_at": "2026-08-28T04:04:49.717961+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/jraph"}], "taxonomy_version": 1}, "enriched": {"inputs": [], "kind": "computed", "method": "enrichment_status"}, "function": {"confidence": null, "enriched_at": "2026-08-30T04:34:33.644239+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "c74ad85491a11353909917d1213202999599f0e3734009755ba6c0d3b87d546a", "fetched_at": "2026-08-28T04:04:49.717961+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/jraph"}], "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:04:49.717961+00:00", "source": "github"}, "language": {"kind": "observed", "observed_at": "2026-08-28T04:04:49.717961+00:00", "source": "github"}, "license": {"kind": "observed", "observed_at": "2026-08-28T04:04:49.717961+00:00", "source": "github"}, "license_family": {"inputs": ["license"], "kind": "computed", "method": "license_family"}, "maturity": {"confidence": null, "enriched_at": "2026-08-30T04:34:33.644239+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "c74ad85491a11353909917d1213202999599f0e3734009755ba6c0d3b87d546a", "fetched_at": "2026-08-28T04:04:49.717961+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/jraph"}], "taxonomy_version": 1}, "member_repos": {"kind": "observed", "observed_at": "2026-08-28T04:04:49.717961+00:00", "source": "github"}, "name": {"kind": "observed", "observed_at": "2026-08-28T04:04:49.717961+00:00", "source": "github"}, "platform": {"confidence": null, "enriched_at": "2026-08-30T04:34:33.644239+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "c74ad85491a11353909917d1213202999599f0e3734009755ba6c0d3b87d546a", "fetched_at": "2026-08-28T04:04:49.717961+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/jraph"}], "taxonomy_version": 1}, "pushed_at": {"kind": "observed", "observed_at": "2026-08-28T04:04:49.717961+00:00", "source": "github"}, "repo": {"kind": "observed", "observed_at": "2026-08-28T04:04:49.717961+00:00", "source": "github"}, "stars": {"kind": "observed", "observed_at": "2026-08-28T04:04:49.717961+00:00", "source": "github"}, "tags": {"confidence": null, "enriched_at": "2026-08-30T04:34:33.644239+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "c74ad85491a11353909917d1213202999599f0e3734009755ba6c0d3b87d546a", "fetched_at": "2026-08-28T04:04:49.717961+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/jraph"}], "taxonomy_version": 1}, "topics": {"kind": "observed", "observed_at": "2026-08-28T04:04:49.717961+00:00", "source": "github"}, "urls": {"kind": "observed", "observed_at": "2026-08-28T04:04:49.717961+00:00", "source": "github"}, "use_cases": {"confidence": null, "enriched_at": "2026-08-30T04:34:33.644239+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "c74ad85491a11353909917d1213202999599f0e3734009755ba6c0d3b87d546a", "fetched_at": "2026-08-28T04:04:49.717961+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/jraph"}], "taxonomy_version": 1}, "what_it_is": {"confidence": null, "enriched_at": "2026-08-30T04:34:33.644239+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "c74ad85491a11353909917d1213202999599f0e3734009755ba6c0d3b87d546a", "fetched_at": "2026-08-28T04:04:49.717961+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/jraph"}], "taxonomy_version": 1}, "when_to_avoid": {"confidence": null, "enriched_at": "2026-08-30T04:34:33.644239+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "c74ad85491a11353909917d1213202999599f0e3734009755ba6c0d3b87d546a", "fetched_at": "2026-08-28T04:04:49.717961+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/jraph"}], "taxonomy_version": 1}, "when_to_choose": {"confidence": null, "enriched_at": "2026-08-30T04:34:33.644239+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "c74ad85491a11353909917d1213202999599f0e3734009755ba6c0d3b87d546a", "fetched_at": "2026-08-28T04:04:49.717961+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/jraph"}], "taxonomy_version": 1}}, "score": {"components": {"activity": 0, "longevity": 100, "rhythm": 8}, "computed_at": "2026-09-02T17:46:02.011165+00:00", "flags": ["archived"], "formula": "round(0.45*activity + 0.35*rhythm + 0.20*longevity); archived -> min(score, 10)", "inputs": {"age_days": 2109, "days_push": 898, "days_rel": null, "gap_med": null, "n_releases_24m": 0}, "score": 10, "version": 2}, "staleness": {"enrichment_outdated": false, "low_confidence": false, "scrape_days": 9, "stale_scrape": false}}