AutoML-Agent
Open-source multi-agent LLM framework for end-to-end automated machine learning pipelines.
🛡️ AgentReady threat assessment
MAESTRO 7-layer threat model + OWASP AIVSS risk score for AutoML-Agent, derived from its capabilities.
AIVSS 8.6 · High
View MAESTRO 7-layer threat model →Overview
AutoML-Agent is an open-source multi-agent framework that uses LLMs to automate the full ML pipeline—from data ingestion to model deployment. Designed for diverse data modalities, it decomposes tasks across specialized agents (e.g. prompt, data, model, operation), with retrieval‑augmented planning and multi-stage verification to ensure robust and deployable model solutions.
Key features
- AutoML
- multi-agent
- LLM orchestration
- end-to-end pipeline
- open-source
Use cases
- Automating full ML workflows with natural‑language instructions.
- Building end‑to‑end AI models across vision, NLP, tabular, time‑series, graph.
- Leveraging modular agents for data, modeling, deployment orchestration.
- Ensuring plan quality via retrieval‑augmented planning and verification.