openai/circuit_sparsity
Open-source release accompanying Gao et al. 2025
What it solves
This project provides tools to inspect and visualize sparse circuit models, helping researchers understand how neural networks perform specific tasks by identifying the minimal set of components (circuits) required for those tasks.
How it works
The project implements a lightweight GPT architecture for inference and a Streamlit-based dashboard. The visualizer loads circuit masks, activations, and importance data from a remote blob store, allowing users to interactively explore how pruning sweeps and node budgets affect model behavior.
Who it’s for
Researchers and developers interested in mechanistic interpretability and the study of sparse circuits in neural networks.
Highlights
- Interactive Streamlit dashboard with Plotly-rendered plots for circuit exploration.
- Lightweight GPT implementation optimized for CPU/GPU inference.
- Access to a suite of released models (including
csp_yoloandcsp_bridgevariants) and pruning sweep data. - Built-in utilities for recording activations via hooks and managing cached data artifacts.
Related
- Project
- Project
- Project
- Project