distributed-llm-pretraining-torchtitan
TorchTitan 是 PyTorch 官方的大规模 LLM 预训练平台,提供 PyTorch 原生实现的 4D 并行能力(FSDP2、TP、PP、CP),据资料称在 H100 GPU 上相较基线可获得 65% 以上加速。该技能围绕 TorchTitan 提供预训练 Llama 3.1、DeepSeek V3 等模型的安装、配置与启动指导,涵盖单节点训练、SLURM 多节点训练、Float8 训练以及 405B 级模型的 4D 并行流程,并包含检查点、监控与常见问题排查。适合需要从零预训练 8B 至 405B 以上模型、并希望使用无第三方依赖的 PyTorch 原生方案的场景。