Skip to main content

Chrome DevTools for AI models and agents — see attention, features, and agent steps inside any local model.

Project description

ModelMRI

Chrome DevTools for AI models and agents. Load any local model — LLM, VLM, or robot policy — and see inside it while it runs: attention, features, circuits, agent steps. Local-first. MIT.

🚧 Under active construction, in public. v0.1 (live attention playground) ships in ~2 weeks. Follow the build: Substack

Why

When a model gives a wrong answer, you can't see why. When an agent fails at step 47, you get a wall of logs. When a robot policy drops the object, you get nothing. The research tools that can see inside (SAEs, attention analysis, circuit tracing) live in notebooks only specialists can drive.

ModelMRI packages that research into a tool with the ergonomics of browser DevTools: one-line install, open localhost:5900, look inside.

Planned (the 12-week public roadmap)

  • v0.1 — load a HuggingFace LLM, watch attention flow live at 60fps (WebGL)
  • v0.2 — SAE feature browser: see the concepts inside the model, steer them
  • v0.3 — agent mode: record any Anthropic-SDK/Claude-Code agent run, replay it, find the failing step
  • v0.4 — the first interactive tool for looking inside a robot policy (SmolVLA + LeRobot)
  • v0.5 — polish, zero-install hosted demo, launch

Install

pip install modelmri
modelmri serve               # then open http://localhost:5900

Or from source (dev):

git clone https://github.com/muhammadmahadazher/ModelMRI && cd ModelMRI
cd frontend && npm install && npm run build && cd ..
uv sync
uv run modelmri serve

Click Load Qwen2.5-0.5B-Instruct (~1 GB one-time download), type a prompt, watch tokens stream — then hover any token to see what the model attended to.

Status

Piece State
FastAPI backend: model loading, REST, WebSocket token streaming ✅ working
Built-in playground page (temporary, pre-React) ✅ working
Attention inspector: hover any token, see what it attended to ✅ working
SAE feature browser 🏗️ next
Frontend (React + WebGL) 🏗️ next
pip install modelmri placeholder published (real release at v0.1)

MIT © Muhammad Mahad Azher

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

modelmri-0.1.0.tar.gz (81.7 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

modelmri-0.1.0-py3-none-any.whl (62.6 kB view details)

Uploaded Python 3

File details

Details for the file modelmri-0.1.0.tar.gz.

File metadata

  • Download URL: modelmri-0.1.0.tar.gz
  • Upload date:
  • Size: 81.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.10.0 {"installer":{"name":"uv","version":"0.10.0","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":null,"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for modelmri-0.1.0.tar.gz
Algorithm Hash digest
SHA256 553913bdbfa664cfd151608bbf3f55a10ffb1106531b837a5cad8cfbb315d742
MD5 ac503b968b39dc17f8552fa6c2177821
BLAKE2b-256 1109f0c32f497d90980d41fdfe16a564f97ec72f948f91ed6b0795cd90e221eb

See more details on using hashes here.

File details

Details for the file modelmri-0.1.0-py3-none-any.whl.

File metadata

  • Download URL: modelmri-0.1.0-py3-none-any.whl
  • Upload date:
  • Size: 62.6 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.10.0 {"installer":{"name":"uv","version":"0.10.0","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":null,"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for modelmri-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 36e8e15e403a690eb091a4f2081d411b8113a5d7e7d772ed880e3f293beedb57
MD5 f6264a588641d95574ae4397af607894
BLAKE2b-256 3cc4125d3fe0a22e0f4d1c33f4a201049444f65081a5a4046170dec6ed9223be

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page