train-llm-from-scratch
用PyTorch从零训练自己的Transformer语言模型的分步教程,涵盖从原始文本到生成和对齐。适合想了解大模型原理的学习者。
挑一个能复现的小 issue,用 AI 编程助手把它修好。
分享到 X许可证: MIT
概述
这是一套基于《Attention Is All You Need》论文、用PyTorch训练自己的Transformer语言模型的分步教程和脚本。内容从原始文本和分词开始,得到基础模型,再依次进行监督微调、奖励模型、PPO、DPO和GRPO,最后是评估与对话。算法全部用原生PyTorch手写,不依赖trl、peft或transformers。作者称用单张GPU即可训练百万或十亿参数规模的模型。
主要特点
- 用原生PyTorch从零实现Transformer
- 从原始文本到预训练基础模型的完整流程
- 涵盖SFT、奖励模型、PPO、DPO和GRPO
- 不依赖trl、peft和transformers
- 最后包含评估与对话
适用场景
想通过亲手阅读和运行每一步来理解语言模型如何构建与对齐的学习者。这是教学项目,示例输出来自一个约1300万参数的小模型。
- 上游仓库
- FareedKhan-dev/train-llm-from-scratch
- GitHub 上的 fork
- Guo-astro/train-llm-from-scratch
- 上游 Star 数
- 12k
- 分类
- AI 智能体与大模型工具
- 编程语言
- Python
- 许可证
- MIT
- Fork 时间
- 2026-09-25
- 同步状态
- 已同步上次同步 2026-10-19
同分类更多: AI 智能体与大模型工具
为研究者提供从想法评估、文献调研到论文写作与投稿前检查的结构化 AI 辅助,弥补通用 AI 在科研判断上的不足。
Fork 时间 2026-10-18上次同步 2026-10-19自定义许可证(见仓库)AI 智能体与大模型工具GitHub
RLVE为语言模型强化学习提供可调整难度、可验证的环境,用于生成问题并评估模型输出。
Fork 时间 2026-10-17上次同步 2026-10-19许可证: MITAI 智能体与大模型工具GitHub
为 AI 代理提供模块化的 Android 开发指导,帮助其执行专业工作流程。可使用 Android CLI 安装单个技能或全部技能。
Fork 时间 2026-10-16上次同步 2026-10-19许可证: Apache-2.0AI 智能体与大模型工具GitHub
Pi Agent 是适用于 macOS、Windows 和 Linux 的本地 AI 编程桌面应用,无需配置终端环境即可管理对话、项目文件并运行已保存的提示词。
Fork 时间 2026-10-15上次同步 2026-10-19许可证: MITAI 智能体与大模型工具GitHub