{"adoption": {"forks": 70, "observed_at": "2026-08-28T04:05:51.649951+00:00", "stars": 1901}, "canonical_url": "https://ross.abutalabs.com/products/penzai", "card": {"archived": true, "artifact_type": "library", "description": "A JAX research toolkit for building, editing, and visualizing neural networks.", "domain": ["machine-learning", "deep-learning", "large-language-models", "developer-tools"], "enriched": true, "function": ["machine-learning", "deep-learning", "data-visualization", "developer-tools"], "health_score": 10, "homepage": "https://penzai.readthedocs.io/", "language": "Python", "license": "Apache-2.0", "license_family": "permissive", "maturity": "active", "member_repos": ["google-deepmind/penzai"], "name": "google-deepmind/penzai", "platform": ["python", "jvm"], "pushed_at": "2025-06-22T05:45:13+00:00", "repo": "google-deepmind/penzai", "stars": 1901, "tags": ["jax", "interpretability", "model-surgery", "transformers", "pytrees", "neural-networks", "fine-tuning", "research-toolkit"], "topics": ["fine-tuning", "interpretability", "jax", "neural-networks", "visualization"], "urls": [], "use_cases": ["reverse-engineer and ablate components of trained neural networks", "inspect and probe internal activations of transformer models", "perform model surgery on pretrained Gemma, Llama, Mistral, or GPT-NeoX models", "visualize deeply nested JAX pytrees and NDArrays in notebooks", "build and train custom neural network architectures declaratively", "debug neural network architectures interactively", "fine-tune large language models with editable structures"], "what_it_is": "Penzai is a JAX research toolkit for building, editing, and visualizing neural networks as legible, functional pytree data structures. It includes tools for model surgery, interpretability research, activation inspection, and interactive visualization of deeply nested model structures.", "when_to_avoid": ["you need a production training framework with mature ecosystem like PyTorch", "you want a simple high-level API like Keras for standard workflows", "your project is not based on JAX or the Python ML ecosystem", "you need distributed large-scale training infrastructure out of the box"], "when_to_choose": ["you need to modify or analyze trained models rather than just train them", "you want interpretability research tooling for transformer LLMs in JAX", "you prefer models as inspectable, editable pytree data structures", "you need interactive visualization of model internals in IPython or Colab"]}, "data_as_of": "2026-08-30T08:39:29.467469+00:00", "members": [{"path": "/products/penzai", "repo": "google-deepmind/penzai", "role": "main", "score": 10}], "provenance": {"archived": {"kind": "observed", "observed_at": "2026-08-28T04:05:51.649951+00:00", "source": "github"}, "artifact_type": {"confidence": null, "enriched_at": "2026-08-30T03:11:56.543876+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "0a65114c31e930007dd26ea3f276f63ac8d03bba92779ef903d75d162a38c63c", "fetched_at": "2026-08-28T04:05:51.649951+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/penzai"}, {"content_hash": "e9df2f652c40d5aa5e2191800f205eef82160aee559f799731e6524e369a4543", "fetched_at": "2026-08-29T10:51:35.332162+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/penzai/json"}], "taxonomy_version": 1}, "description": {"kind": "observed", "observed_at": "2026-08-28T04:05:51.649951+00:00", "source": "github"}, "domain": {"confidence": null, "enriched_at": "2026-08-30T03:11:56.543876+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "0a65114c31e930007dd26ea3f276f63ac8d03bba92779ef903d75d162a38c63c", "fetched_at": "2026-08-28T04:05:51.649951+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/penzai"}, {"content_hash": "e9df2f652c40d5aa5e2191800f205eef82160aee559f799731e6524e369a4543", "fetched_at": "2026-08-29T10:51:35.332162+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/penzai/json"}], "taxonomy_version": 1}, "enriched": {"inputs": [], "kind": "computed", "method": "enrichment_status"}, "function": {"confidence": null, "enriched_at": "2026-08-30T03:11:56.543876+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "0a65114c31e930007dd26ea3f276f63ac8d03bba92779ef903d75d162a38c63c", "fetched_at": "2026-08-28T04:05:51.649951+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/penzai"}, {"content_hash": "e9df2f652c40d5aa5e2191800f205eef82160aee559f799731e6524e369a4543", "fetched_at": "2026-08-29T10:51:35.332162+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/penzai/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:05:51.649951+00:00", "source": "github"}, "language": {"kind": "observed", "observed_at": "2026-08-28T04:05:51.649951+00:00", "source": "github"}, "license": {"kind": "observed", "observed_at": "2026-08-28T04:05:51.649951+00:00", "source": "github"}, "license_family": {"inputs": ["license"], "kind": "computed", "method": "license_family"}, "maturity": {"confidence": null, "enriched_at": "2026-08-30T03:11:56.543876+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "0a65114c31e930007dd26ea3f276f63ac8d03bba92779ef903d75d162a38c63c", "fetched_at": "2026-08-28T04:05:51.649951+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/penzai"}, {"content_hash": "e9df2f652c40d5aa5e2191800f205eef82160aee559f799731e6524e369a4543", "fetched_at": "2026-08-29T10:51:35.332162+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/penzai/json"}], "taxonomy_version": 1}, "member_repos": {"kind": "observed", "observed_at": "2026-08-28T04:05:51.649951+00:00", "source": "github"}, "name": {"kind": "observed", "observed_at": "2026-08-28T04:05:51.649951+00:00", "source": "github"}, "platform": {"confidence": null, "enriched_at": "2026-08-30T03:11:56.543876+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "0a65114c31e930007dd26ea3f276f63ac8d03bba92779ef903d75d162a38c63c", "fetched_at": "2026-08-28T04:05:51.649951+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/penzai"}, {"content_hash": "e9df2f652c40d5aa5e2191800f205eef82160aee559f799731e6524e369a4543", "fetched_at": "2026-08-29T10:51:35.332162+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/penzai/json"}], "taxonomy_version": 1}, "pushed_at": {"kind": "observed", "observed_at": "2026-08-28T04:05:51.649951+00:00", "source": "github"}, "repo": {"kind": "observed", "observed_at": "2026-08-28T04:05:51.649951+00:00", "source": "github"}, "stars": {"kind": "observed", "observed_at": "2026-08-28T04:05:51.649951+00:00", "source": "github"}, "tags": {"confidence": null, "enriched_at": "2026-08-30T03:11:56.543876+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "0a65114c31e930007dd26ea3f276f63ac8d03bba92779ef903d75d162a38c63c", "fetched_at": "2026-08-28T04:05:51.649951+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/penzai"}, {"content_hash": "e9df2f652c40d5aa5e2191800f205eef82160aee559f799731e6524e369a4543", "fetched_at": "2026-08-29T10:51:35.332162+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/penzai/json"}], "taxonomy_version": 1}, "topics": {"kind": "observed", "observed_at": "2026-08-28T04:05:51.649951+00:00", "source": "github"}, "urls": {"kind": "observed", "observed_at": "2026-08-28T04:05:51.649951+00:00", "source": "github"}, "use_cases": {"confidence": null, "enriched_at": "2026-08-30T03:11:56.543876+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "0a65114c31e930007dd26ea3f276f63ac8d03bba92779ef903d75d162a38c63c", "fetched_at": "2026-08-28T04:05:51.649951+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/penzai"}, {"content_hash": "e9df2f652c40d5aa5e2191800f205eef82160aee559f799731e6524e369a4543", "fetched_at": "2026-08-29T10:51:35.332162+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/penzai/json"}], "taxonomy_version": 1}, "what_it_is": {"confidence": null, "enriched_at": "2026-08-30T03:11:56.543876+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "0a65114c31e930007dd26ea3f276f63ac8d03bba92779ef903d75d162a38c63c", "fetched_at": "2026-08-28T04:05:51.649951+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/penzai"}, {"content_hash": "e9df2f652c40d5aa5e2191800f205eef82160aee559f799731e6524e369a4543", "fetched_at": "2026-08-29T10:51:35.332162+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/penzai/json"}], "taxonomy_version": 1}, "when_to_avoid": {"confidence": null, "enriched_at": "2026-08-30T03:11:56.543876+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "0a65114c31e930007dd26ea3f276f63ac8d03bba92779ef903d75d162a38c63c", "fetched_at": "2026-08-28T04:05:51.649951+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/penzai"}, {"content_hash": "e9df2f652c40d5aa5e2191800f205eef82160aee559f799731e6524e369a4543", "fetched_at": "2026-08-29T10:51:35.332162+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/penzai/json"}], "taxonomy_version": 1}, "when_to_choose": {"confidence": null, "enriched_at": "2026-08-30T03:11:56.543876+00:00", "kind": "inferred", "prompt_version": 1, "sources": [{"content_hash": "0a65114c31e930007dd26ea3f276f63ac8d03bba92779ef903d75d162a38c63c", "fetched_at": "2026-08-28T04:05:51.649951+00:00", "kind": "readme", "missing": false, "url": "https://github.com/google-deepmind/penzai"}, {"content_hash": "e9df2f652c40d5aa5e2191800f205eef82160aee559f799731e6524e369a4543", "fetched_at": "2026-08-29T10:51:35.332162+00:00", "kind": "registry_pypi", "missing": false, "url": "https://pypi.org/pypi/penzai/json"}], "taxonomy_version": 1}}, "score": {"components": {"activity": 28, "longevity": 63, "rhythm": 28}, "computed_at": "2026-09-03T02:20:16.233290+00:00", "flags": ["archived"], "formula": "round(0.45*activity + 0.35*rhythm + 0.20*longevity); archived -> min(score, 10)", "inputs": {"age_days": 882, "days_push": 437, "days_rel": 512, "gap_med": 71, "n_releases_24m": 4}, "score": 10, "version": 2}, "staleness": {"enrichment_outdated": false, "low_confidence": false, "scrape_days": 9, "stale_scrape": false}}