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_yolo and csp_bridge variants) and pruning sweep data.
  • Built-in utilities for recording activations via hooks and managing cached data artifacts.

Related

  • Project
  • Project
  • Project
  • Project