MLE-Agent

What does this agent do, and when should you use it?

MLE-Agent is presented as a pairing LLM agent for machine learning engineers and researchers. Its `mle` command-line interface creates and starts project directories and includes interactive terminal chat. The repository describes baseline construction, code execution and debugging on a local machine or cloud environment, plus research retrieval through Arxiv, Papers with Code, and general keyword search. It also documents Kaggle workflows, weekly-report generation, and local RAG support, while listing OpenAI GPTs, Anthropic Claude 3.5 Sonnet, and Ollama Llama3 among supported models. It fits ML teams that want one project-local workflow for planning, research, execution, and reporting, but the supplied material does not document provider credentials or model configuration.

After mle new <project name> creates a directory, mle start begins the ML/AI project workflow and mle chat opens an interactive terminal session. The project describes building ML/AI baselines from requirements, retrieving methods through Arxiv, Papers with Code, and general keyword search, and executing, debugging, and fixing code on a local machine or cloud environment. mle kaggle starts a Kaggle task, while mle kaggle --auto accepts dataset paths, a description, submission file, submission example, and competition ID for an automated workflow spanning data preparation, training, and debugging. For reporting, mle report starts a local web application at http://localhost:3000/ for GitHub-based reports, and mle report-local --email=<git email> --start-date=YYYY-MM-DD --end-date=YYYY-MM-DD <path_to_git_repo> creates a report from a local Git repository.

1. An ML engineer who has a vague request, such as predicting stock prices from historical data, and wants to prototype and test a baseline locally.
2. A Kaggle participant who has joined a competition and has the datasets and submission artifacts ready for the parameterized mle kaggle --auto workflow.
3. A researcher planning an ML approach who wants to consult Arxiv, Papers with Code, and keyword-search results alongside implementation work.
4. A developer maintaining an existing ML project who wants a terminal-based mle chat interaction mode within that project directory.
5. A team member who needs a weekly work report covering development progress, communication notes, references, and to-do items from GitHub or a local Git repository.

How do you install or deploy this agent?

Install from PyPI:

pip install -U mle-agent

Or with uv:

uv pip install -U mle-agent

Install from source:

git clone https://github.com/MLSysOps/MLE-agent.git
cd MLE-agent
uv venv .venv
source .venv/bin/activate
pip install -e .

The supplied material lists support for providers including OpenAI, Anthropic, and Ollama, but does not document required credentials, environment variables, or model-configuration steps.

How do you use this agent?

Create and start a project:

mle new <project name>
cd <project name>
mle start

Start terminal chat in that project:

mle chat

Start a Kaggle workflow:

mle kaggle

Generate a report from a local Git repository:

mle report-local --email=<git email> --start-date=YYYY-MM-DD --end-date=YYYY-MM-DD <path_to_git_repo>

What are this agent's strengths and limitations?

How does this agent compare with similar options?

Key facts side by side with the most closely related agents.

How does FollowAgents rate this agent?

FAQ