ColossalAI

中文 Tutorial 入门书

📌 版本基线: hpcaitech/ColossalAI · main · commit 4f9953be335e · 详见 版本基线表

面向:希望用国产框架完成大模型训练、对"低成本大模型"路线感兴趣的工程师。

前置:会写 PyTorch;理解 ZeRO / TP / PP 的基本概念。

本书的目标

目录

章节标题状态
第 1 章 项目背景与定位
第 2 章 安装与依赖(含 CUDA extension 编译)
第 3 章 快速上手:从单卡到 8 卡的最小例子
第 4 章 核心概念:Booster、Plugin、ShardFormer
第 5 章 Gemini:异构显存 + CPU 卸载
第 6 章 HybridParallelPlugin:一份配置开多种并行
第 7 章 ShardFormer:自动按 layer 切模型
第 8 章 实战:用 Colossal-LLaMA 训中文增强模型
第 9 章 实战:ColossalChat 的 SFT → RM → PPO 流水线
第 10 章性能优化:fused kernel / chunked communication
第 11 章调试与常见错误
第 12 章Checkpoint 与 HF 互转
第 13 章源码导读与社区生态
📁 配套源码:../../repos/colossalai/  ·  🔗 上一级:总目录