distributed-llm-pretraining-torchtitan

SKILLWorkflowcommunity
v0.0.0Orchestra-ResearchMITUpdated 2mo agoSource →

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.

Community-submitted skill. Not yet reviewed by the Forge team. Full prompt content may not be available.Request review →
12kRepo stars
1Clients
1Formats
2mo agoLast update
Skill
AuthorOrchestra-Research
Version0.0.0
LicenseMIT
CategoryWorkflow
Formatsskill.md
PromptOpen (see Prompt tab)
Compatibility
Claude✓ Supported
Cursor
Copilot
ChatGPT
Gemini
About

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.

Keywords
skillclaude