12.5K 星 GitHub 仓库:纯 PyTorch 手写从预训练到 GRPO 全流程

techNmak · x · 2026-10-27

一个 12.5K 星的 GitHub 仓库「Train LLM From Scratch」把整门 LLM 工程课装进了一个代码库:从原始文本出发,用纯 PyTorch 手写 Transformer,一路覆盖到强化学习,不依赖 trl、peft、transformers。

仓库实现包括:

各阶段共享同一个底层 Transformer,可对照观察训练目标、奖励计算与优化循环的演变;另有现代架构版(RoPE、RMSNorm、SwiGLU、GQA、MoE、KV cache)以及 Muon 优化器、LoRA、SimPO、投机解码、int8 权重等新技巧,单 GPU 甚至笔记本 CPU 即可训练。

原文链接 →

「编程与Agent」频道最新

更多「编程与Agent」频道 AI 资讯 →