WecoAI/aideml

Health v2 · maintenance only

Adoption not part of the score

1494 stars · 221 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

AIDE ML is an open-source Python package implementing an LLM-driven agent that uses agentic tree search to autonomously write, debug, and improve machine-learning code against a user-defined metric. It is the research reference build behind Weco and was referenced in OpenAI's MLE-bench.

Use cases

• automatically build ML pipelines from a dataset and a natural-language goal
• enter Kaggle-style competitions with an autonomous AI data scientist
• research new agent architectures by swapping search heuristics or LLM backends
• optimize code against a measurable metric using tree search
• replicate the AIDE paper's results on MLE-bench
• prototype predictive models without hand-tuning

When to choose

• you want an autonomous agent to draft and iteratively improve ML code for a dataset
• you're an agent-architecture researcher wanting an extensible tree-search baseline
• you need a research-friendly, pip-installable implementation of the AIDE algorithm

When to avoid

• you need a production ML platform with experiment tracking and dashboards - use the Weco product instead
• your task has no quantifiable evaluation metric
• you want a no-code AutoML tool with a GUI

Facets

library · maturity active

agent-framework machine-learning llm-inference cli data-science machine-learning artificial-intelligence data-science python cli cross-platform llm-agent tree-search automated-machine-learning kaggle autoresearch code-optimization mle-bench ai-agents automation

Member repositories

For agents

markdown · JSON · MCP: product_card(name="WecoAI/aideml")