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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
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
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
553913bdbfa664cfd151608bbf3f55a10ffb1106531b837a5cad8cfbb315d742
|
|
| MD5 |
ac503b968b39dc17f8552fa6c2177821
|
|
| BLAKE2b-256 |
1109f0c32f497d90980d41fdfe16a564f97ec72f948f91ed6b0795cd90e221eb
|
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
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
36e8e15e403a690eb091a4f2081d411b8113a5d7e7d772ed880e3f293beedb57
|
|
| MD5 |
f6264a588641d95574ae4397af607894
|
|
| BLAKE2b-256 |
3cc4125d3fe0a22e0f4d1c33f4a201049444f65081a5a4046170dec6ed9223be
|