Autoresearch是一项完全由 AI 驱动的自主机器学习研究实验由 Andrej Karpathy创建。其核心思想非常简单赋予 Agent 一个真实可用的 GPT 训练环境让其自主进行实验——修改代码、运行 5 分钟的短时训练、评估结果并决定保留还是丢弃每次更改。当你入睡后到第二天早上该 Agent 可能已经完成了约 100 次实验每一次都是旨在降低验证损失的真诚尝试。这不是一个玩具演示——而是一个极简但严肃的预训练研究环境在这里“研究员”永不眠息。整个代码库刻意保持极简——只有三个核心文件、少量辅助文件并且零配置框架。这并非偶然。极简即是架构。通过将问题限制在单 GPU 上的单个可编辑文件和单一指标该项目消除了基础设施带来的开销使 Agent或你能够在固定的时间预算内完全专注于如何更好地训练 transformer。宏观架构该系统围绕_固定_部分与_可变_部分之间的清晰职责分离而构建。理解这一边界是掌握 autoresearch 其他一切内容的关键。绿色组件由人类编写。蓝色组件是不可变的基础设施。橙色组件是 Agent 的试验场。紫色的 Agent 负责编排实验循环而灰色的输出则记录了发生的一切。项目结构代码库的结构印证了其架构的简洁性。每个文件都有明确的职责且整个代码库中有意义的 Python 代码不足 1000 行。autoresearch/├── prepare.py ├── train.py # ✏️ EDITABLE — GPT model, optimizer, training loop├── program.md # AGENT PROMPT — Instructions for the AI researcher├── pyproject.toml # CONFIG — Project dependencies (uv-managed)├── analysis.ipynb # ANALYSIS — Notebook for inspecting experiment results├── progress.png # VISUAL — Results chart from a sample overnight run├── README.md # DOCS — Project readme└── uv.lock # LOCK — Pinned dependency versions这三个关键文件映射了一个三角色的设计文件所属方用途可否修改prepare.py人类固定下载数据训练 BPE 分词器提供数据加载器和evaluate_bpb()❌ 绝不修改train.pyAI Agent包含 GPT 模型架构、MuonAdamW 优化器、超参数和训练循环✅ 可以——这是沙盒program.md人类动态演进向 Agent 提供设置说明、实验协议、日志格式和规则✅ 仅限人类修改实验循环如何运作一旦 Agent 启动它就会进入一个无限循环这镜像了人类机器学习研究人员的可能工作方式——但速度更快且不知疲倦。每个周期大约需要 5 分钟。Agent 从不暂停以征求许可。一旦循环开始它就会无限期运行——预期是让你在夜间保持其运行醒来时即可看到大约 100 次已完成的实验所有结果都被追踪记录在results.tsv文件中并且只有在发现改进时git 分支才会向前推进。快速开始在 15 分钟内让你的机器跑通 autoresearch。本指南将带你走完每一步——从环境配置到首次成功的 5 分钟训练运行——以便你在将控制权交给自主 AI agent 之前验证整个流水线能够正常工作。前置条件在开始之前请确保你的环境满足以下要求。Autoresearch 在设计上力求极简但它确实需要支持 CUDA 的 NVIDIA 硬件——整个技术栈都是围绕单 GPU 训练以及 Flash Attention 3 构建的。需求最低要求推荐配置备注GPU任何支持 CUDA 的 NVIDIA GPUH100在 H100 上经过测试其他 GPU 会产生不同的吞吐量数据Python3.103.10通过 pyproject.toml 中的requires-python强制执行uv最新稳定版最新稳定版Python 包管理器——通过以下命令安装磁盘空间~5 GB~20 GB数据分片会被下载到~/.cache/autoresearch/目录下显存~10 GB40 GB默认的 8 层模型在峰值时使用约 44 GB也可以使用更小的配置本项目通过自定义包索引锁定了 PyTorch 2.9.1 与 CUDA 12.8因此你无需单独安装 CUDA——PyTorch 自带 CUDA 运行时。安装配置整个安装过程使用 uv这是一个快速的 Python 包管理器。它在一个工具中集成了依赖解析、虚拟环境管理和 Python 版本管理。按顺序运行以下四条命令curl -LsSf https://astral.sh/uv/install.sh | sh # 2. 同步依赖——创建虚拟环境并安装所有内容uv sync # 3. 下载训练数据并训练 BPE 分词器约 2 分钟uv run prepare.py # 4. 运行你的第一次训练实验约 5 分钟uv run train.py我们来详细拆解每条命令的实际作用因为理解这一流程对于后续排查问题至关重要。第 2 步 ——uv sync会读取 pyproject.toml 并将所有依赖安装到自动创建的虚拟环境中。核心依赖是torch2.9.1它拉取自 NVIDIA 的 CUDA 12.8 wheel 索引。其他包包括kernels用于 Flash Attention 3、rustbpe快速的 BPE 分词器、numpy、pandas、pyarrow用于读取 Parquet 数据分片以及matplotlib用于分析 Notebook。第 3 步 ——uv run prepare.py是一个一次性操作负责执行两项任务。首先它会从 Hugging Face 下载训练数据分片到~/.cache/autoresearch/data/目录。默认情况下它会获取 10 个分片每个约 500 MB这足以让你开始上手后续你可以使用--num-shards -1下载全部 6,542 个分片。其次它会训练一个词表大小为 8,192 个 token 的 BPE 分词器并将其保存到~/.cache/autoresearch/tokenizer/目录。当你在末尾看到打印出Done! Ready to train.时就说明执行成功了。第 4 步 ——uv run train.py会启动完整的训练流水线构建模型精准运行 5 分钟的实际训练时间可在 prepare.py 中通过TIME_BUDGET 300进行配置在保留的验证集分片上进行评估并打印最终摘要。如果该过程顺利结束且无报错说明你的环境已完全配置成功。解读训练输出当uv run train.py执行完毕后你会看到一段启动序列随后是训练进度最后是一个摘要块。以下是各部分的含义启动输出展示了由DEPTH超参数推导出的模型配置。模型维度计算方式为depth × ASPECT_RATIO向上取整为HEAD_DIM 128的倍数注意力头的数量也随之确定。Vocab size: 8,192Model config: GPTConfig(sequence_len2048, vocab_size8192, n_layer8, n_head4, n_kv_head4, n_embd512, window_patternSSSL)最终摘要块——即最关键的输出——如下所示---val_bpb: 0.997900training_seconds: 300.1total_seconds: 325.9peak_vram_mb: 45060.2mfu_percent: 39.80total_tokens_M: 499.6num_steps: 953num_params_M: 50.3depth: 8指标衡量内容关注点val_bpb验证集 bits per byte核心目标。数值越低越好。与词表大小无关。training_seconds实际训练挂钟时间应接近 300即时间预算total_seconds包含启动和评估的总时间通常在 320–360 秒之间peak_vram_mbGPU 显存峰值使用量用于监控是否会发生 OOM 崩溃mfu_percent模型 FLOPs 利用率数值越高 GPU 利用率越好通常在 30–50% 之间total_tokens_M处理的 token 数量百万token 越多 单次实验学习越充分num_steps优化器更新步数计算方式为total_tokens / TOTAL_BATCH_SIZEnum_params_M模型参数量百万随DEPTH的变化而缩放val_bpb是对比实验时唯一重要的指标。它通过目标字节长度对交叉熵进行归一化处理因此更改词表大小例如从 8,192 改为 256不会人为地放大或缩小该数值。这使得架构实验之间具备较高的可比性——这也是该框架的一个核心设计决策。常见问题排查以下是首次运行时最可能遇到的问题及其解决方法症状可能原因解决方法uv sync因 PyTorch 错误失败不支持的 CUDA 版本或非 NVIDIA GPU确保已安装 NVIDIA GPU 及对应驱动prepare.py下载报错网络问题或 Hugging Face 速率限制重试或减少分片数uv run prepare.py --num-shards 4运行train.py时出现 OOM默认模型对于你的 GPU 来说过大在 train.py 中将DEPTH降至 4并将DEVICE_BATCH_SIZE降至 32kernels导入错误非 Hopper 架构 GPU非 H100代码会在 train.py 处自动检测 GPU 算力并选择合适的 Flash Attention 3 变体训练缓慢 / MFU 偏低GPU 未被充分利用检查DEVICE_BATCH_SIZE——如果显存允许则调大该值找不到分词器缓存的FileNotFoundError尚未运行prepare.py请先运行uv run prepare.py学AI大模型的正确顺序千万不要搞错了2026年AI风口已来各行各业的AI渗透肉眼可见超多公司要么转型做AI相关产品要么高薪挖AI技术人才机遇直接摆在眼前有往AI方向发展或者本身有后端编程基础的朋友直接冲AI大模型应用开发转岗超合适就算暂时不打算转岗了解大模型、RAG、Prompt、Agent这些热门概念能上手做简单项目也绝对是求职加分王给大家整理了超全最新的AI大模型应用开发学习清单和资料手把手帮你快速入门学习路线:✅大模型基础认知—大模型核心原理、发展历程、主流模型GPT、文心一言等特点解析✅核心技术模块—RAG检索增强生成、Prompt工程实战、Agent智能体开发逻辑✅开发基础能力—Python进阶、API接口调用、大模型开发框架LangChain等实操✅应用场景开发—智能问答系统、企业知识库、AIGC内容生成工具、行业定制化大模型应用✅项目落地流程—需求拆解、技术选型、模型调优、测试上线、运维迭代✅面试求职冲刺—岗位JD解析、简历AI项目包装、高频面试题汇总、模拟面经以上6大模块看似清晰好上手实则每个部分都有扎实的核心内容需要吃透我把大模型的学习全流程已经整理好了抓住AI时代风口轻松解锁职业新可能希望大家都能把握机遇实现薪资/职业跃迁这份完整版的大模型 AI 学习资料已经上传CSDN朋友们如果需要可以微信扫描下方CSDN官方认证二维码免费领取【保证100%免费】