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

by ihatesea69· Repository·other
Also installable via skills CLI
npx skills add ihatesea69/HieuNghi-AI-Skills/airesearch_skills/01-model-architecture/torchtitan

Source

Path:airesearch_skills/01-model-architecture/torchtitan/SKILL.md(main)

Related in other

torchtitan | AgentArea Skills