superml

SuperML

Give your AI coding agent ML engineering superpowers.

Watch how SuperML works in 90 seconds:

If SuperML helps you, give it a ⭐ it helps others find the project.

---

It adds two things your coding agent doesn't have:

ML Pipeline: Seven skills that encode the workflow you already follow. Plan against real framework docs. Catch config mistakes before they cost you GPU hours. Debug OOM, NaN, and divergence by root cause, not by guessing. Get ranked next steps when metrics plateau. An agentic experiment memory carries hypotheses, results, and lessons across sessions — your agent stops repeating failed experiments and starts compounding what works.

Memory: Backed by Leeroopedia, 27k+ pages across 1000+ ML/AI frameworks. Config references, debugging heuristics, implementation patterns, and battle-tested defaults from vLLM to DeepSpeed to LangChain. Built by the Leeroo continuous learning system, structured as a browsable wiki, and continuously updated by AI and human engineers. When your agent recommends a config, it points to the page it learned it from.

Works with Claude Code, Cursor, Codex, OpenCode, and Gemini CLI.

How It Works

1. A session hook loads automatically, zero setup per conversation.
2. Skills guide the ML workflow, verify before launch, debug by root cause, iterate on results, track what worked.
3. MCP tools connect to the Leeroopedia knowledge base, your agent looks things up and cites real docs instead of guessing.
4. A persistent ML agent (ml-expert) handles deeper tasks and remembers your hardware, experiments, and lessons across sessions.

Results

37 ML tasks scored head-to-head: Cursor / Claude Code with SuperML vs without. Each response rated by independent LLM judges across correctness, specificity, mistake prevention, actionability, and grounding.

See TESTED_TASKS.md for detailed scores and methodology.

Prerequisites

API Key (optional, highly recommended)

The plugin works without an API key — skills use web search to ground answers. With a key, your agent gets access to the Leeroopedia knowledge base (27k+ pages, faster and more precise lookups). The plugin will tell you if it's running without a key.

To get a key: app.leeroopedia.com — $20 free credit on signup, no credit card.

export LEEROOPEDIA_API_KEY=kpsk_your_key_here

Add to your shell profile (~/.bashrc, ~/.zshrc) so it persists.

Installation

Claude Code

Register the marketplace, then install the plugin:

/plugin marketplace add leeroo-ai/leeroo-marketplace
/plugin install superml@leeroo-marketplace

Or install directly from GitHub:

claude plugin add --from-github leeroo-ai/superml

Cursor

In Cursor Agent chat (waiting for Cursor team approval):

/add-plugin superml

Or clone into your project — Cursor auto-detects .cursor-plugin/plugin.json:

git clone https://github.com/leeroo-ai/superml.git

Codex

See .codex/INSTALL.md.

OpenCode

See .opencode/INSTALL.md.

Gemini CLI

git clone https://github.com/leeroo-ai/superml.git
gemini extension add ./superml/gemini-extension.json

Alternative: Remote MCP (no local install)

If you just want the knowledge base without the full plugin, see leeroopedia-mcp for setup instructions.

You get the MCP tools (memory) but not the workflow skills (process).

Verify Installation

Start a conversation and try something like: