ZooWork
ZooWork Market · Skill

distributed-llm-pretraining-torchtitan

Provides PyTorch-native distributed LLM pretraining using torchtitan with 4D parallelism (FSDP2, TP, PP, CP). Use when pretraining Llama 3.1, DeepSeek V3, or custom models at scale from 8 to 512+ GPUs with Float8, torch.compile, and distributed checkpointing.

Orchestra-Research
orchestra-research-ai-research-skills-torchtitan · v1.0.0
分类ai-llms
安装次数12
更新时间2026-09-13T07:32:59.143Z
校验状态待验证
Model ArchitectureDistributed TrainingTorchTitanFSDP2Tensor ParallelPipeline ParallelContext ParallelFloat8LlamaPretraining