ml-intern:Hugging Face 开源 ML 工程代理,让 AI 自己读论文训练模型
ml-intern:Hugging Face 开源 ML 工程代理,让 AI 自己读论文训练模型
ml-intern 是 Hugging Face 于 2026 年 4 月开源的自主 ML 工程代理(Autonomous ML Agent)。项目基于自家 smolagents 框架打造,核心能力一句话:给它一个训练目标,它自主完成从文献调研到模型上线的完整 ML 工作流——自己读论文、找代码、挑数据集、写训练脚本、提交流程、推送模型到 Hugging Face Hub,全程无需人工写代码。
功能与原则
ml-intern 的设计理念是”AI 原生工作流”:把机器学习工程师的完整工程能力封装成一个 CLI 工具,用户只需要描述任务目标,代理自主完成剩余全部步骤。核心操作包括:
– 自主研究:查阅 Hugging Face 文档、arXiv 论文、GitHub 代码
– 编写脚本:根据任务自动生成训练代码
– 提交任务:对接 Hugging Face 云端算力提交训练任务
– 模型发布:训练完成后自动推送模型至 Hugging Face Hub
整个过程走 Inference Providers API 计费,不依赖特定模型,支持 Claude、GPT、MiniMax、Kimi、DeepSeek 等多种模型后端。
认可度
截至 2026-08-13,ml-intern 的 GitHub Star 数约为 6,000–7,000 星(2026 年 4 月底首次统计约 5,300 星,日增约 985 星,后续增速趋于平稳)。项目在 2026 年 4 月下旬多次登上 GitHub Trending 日榜,引发 ML 社区广泛关注。Hugging Face 官方背书,smolagents 生态核心项目之一。
链接
GitHub 仓库:https://github.com/huggingface/ml-intern
原作者
由 Hugging Face(大型开源 AI 社区与模型平台)官方团队开发和维护,smolagents 开源 Agent 框架的核心 Demo 项目之一。
介绍
传统机器学习工作流依赖工程师手动完成文献阅读、代码编写、训练调试、模型部署等多个环节,周期长、门槛高。ml-intern 的出现将这一流程彻底自动化。
用户只需在终端输入一句任务描述,如 ml-intern "fine-tune llama on my dataset",系统便会自动分析任务、搜索相关论文和代码实现、编写适配的训练脚本、申请云端 GPU 资源、提交训练任务,最终将微调好的模型上传至个人 Hugging Face Hub。全程可观察、可干预、可中断。
项目采用开放式 Agent 架构,默认通过 bash/read/write/edit 等工具操作本地文件系统;通过 --sandbox-tools 参数可切换至 Hugging Face Space 沙箱环境,远程申请 GPU 算力、创建 HF Space 并执行训练测试。所有会话记录自动以 Claude Code JSONL 格式上传至用户私有 Hugging Face 数据集,方便回溯和分享。
特点
- 全流程自动化:从需求到上线,无需手动写一行训练代码
- 多模型支持:支持 Claude/GPT/MiniMax/Kimi/DeepSeek/Ollama/vLLM 等多种推理后端
- 双工具模式:本地文件系统模式(默认)+ Hugging Face Space 沙箱模式(
--sandbox-tools) - 会话追踪:所有交互自动记录上传至 HF Hub,附 Claude Code JSONL 格式,兼容 Agent Trace Viewer
- Slack 通知集成:支持在审批节点、错误、完成时推送 Slack 消息
- 本地模型兼容:通过
--model ollama/xxx或--model vllm/xxx支持纯本地推理
使用方法
安装步骤:
git clone git@github.com:huggingface/ml-intern.git
cd ml-intern
uv sync
uv tool install -e .
配置环境变量:
在项目根目录创建 .env 文件:
HF_TOKEN=<your-hugging-face-token>
GITHUB_TOKEN=<github-personal-access-token>
(HF_TOKEN 必填,用于 Inference Providers 调用和 Hub 操作;GitHub Token 选填,用于代码搜索)
基本调用示例:
# 默认调用(自动选择模型)
ml-intern "fine-tune llama on my dataset"
# 指定模型
ml-intern --model moonshotai/Kimi-K2.7-Code:novita "train a classifier"
# 使用沙箱工具(远程 GPU 训练)
ml-intern --sandbox-tools "test this training script in a GPU sandbox"
交互式模式下可用 /model 命令切换模型。
使用场景与人群
适用场景:
– 快速验证 ML 想法:给一个任务目标,快速拿到训练好的模型
– 自动化模型训练管线:集成到现有 MLOps 流程
– 跨模型对比:同一个任务用不同模型/数据集快速跑基准测试
– 文献复现:让 AI 自动找论文代码、配置环境、跑出结果
目标用户:
– ML 研究人员:快速验证假设,减少工程负担
– AI 工程师:搭建自动化训练与部署管线
– 独立开发者 / 小团队:没有专职 ML 工程师但有模型微调需求
– 学习者:观察 AI 如何完成完整 ML 工作流,作为学习参考
输入与输出案例
案例 1:微调 Llama
用户输入:
ml-intern "fine-tune llama on my dataset"
代理行为:
1. 分析任务,在 Hugging Face 文档和 arXiv 搜索相关微调方法
2. 查找合适的开源数据集或读取用户指定数据集
3. 编写 PEFT/LoRA 微调脚本
4. 通过 HF Inference Providers 提交训练任务
5. 训练完成后将模型推送至用户 HF Hub
案例 2:零样本分类任务
用户输入:
ml-intern --model ollama/llama3.1:8b "build a sentiment classifier for product reviews"
代理行为:
1. 调研情感分类主流方法(BERT、T5 等)
2. 搜索适合产品评论的数据集(HuggingFace Datasets)
3. 生成训练脚本并用本地 Ollama 模型执行训练
4. 输出模型到本地或推送 HF Hub
评论区
登录后可评论。