提供基于PyTorch原生的分布式大语言模型预训练,使用torchtitan实现4D并行(FSDP2、TP、PP、CP)。适用于预训练Llama 3.1、DeepSeek V3等模型时使用。
npx skills add https://github.com/firecrawl/ai-research-skills --skill distributed-llm-pretraining-torchtitan