Skip to content

distributed-llm-pretraining-torchtitan

from Orchestra-Research/AI-research-SKILLs

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

v1.0.0MIT
359
Lines
1,136
Words
25
Code Blocks

Languages

bashtoml
01-model-architecture/torchtitan/SKILL.md