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 数
- 13k
- 分类
- AI 智能体与大模型工具
- 编程语言
- Python
- 许可证
- MIT
- Fork 时间
- 2026-09-25
- 同步状态
- 已同步上次同步 2026-11-08
同分类更多: AI 智能体与大模型工具
/last30days是一款AI代理技能,可搜索并总结社交平台、视频、新闻等近期信息,帮助用户研究趋势和做准备。
Fork 时间 2026-10-22上次同步 2026-11-08许可证: MITAI 智能体与大模型工具GitHub
这是一个可移植的 C99 推理引擎,无需 GPU 或机器学习框架即可在 CPU 上运行 Kimi K3。它从磁盘读取大型模型检查点,支持在有限内存下进行文本生成或聊天。
Fork 时间 2026-10-21上次同步 2026-11-08许可证: Apache-2.0AI 智能体与大模型工具GitHub
LayerFS在共享历史上为智能体提供可写工作区,支持并行协作。可用它保存更改、创建分支、回退和合并。
Fork 时间 2026-10-21上次同步 2026-11-08许可证: MITAI 智能体与大模型工具GitHub
Magnitude会根据硬件调整内核,在本地运行开放模型,并可连接兼容的智能体或客户端。
Fork 时间 2026-10-19上次同步 2026-11-08许可证: Apache-2.0AI 智能体与大模型工具GitHub