distributed-llm-pretraining-torchtitan

Orchestrate 4D parallelism for large-scale LLM pretraining with TorchTitan.

1|Updated Apr 4, 2026
One-click install
npx skills add https://github.com/m0at/claudemd --skill distributed-llm-pretraining-torchtitan-m0at
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: distributed-llm-pretraining-torchtitan
Source: https://github.com/m0at/claudemd/tree/main/skills/torchtitan
Command: npx skills add https://github.com/m0at/claudemd --skill distributed-llm-pretraining-torchtitan-m0at

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

TorchTitan enables PyTorch-native distributed pretraining of large language models using composable 4D parallelism across multi-GPU clusters.

Core Features & Use Cases

  • 4D parallelism across FSDP2, tensor parallelism (TP), pipeline parallelism (PP), and context parallelism (CP) to scale pretraining.
  • Float8 training on H100 with selective filtering and compile-stage acceleration for speedups.
  • Checkpoint interoperability with HuggingFace and torchtune to save, resume, and convert checkpoints.
  • Multi-node workflows including 8 to 512+ GPU clusters and SLURM-based launches for large-scale pretraining.

Quick Start

Launch a pretraining job on 8 GPUs using a ready config and the training script.

Frequently Asked Questions about distributed-llm-pretraining-torchtitan

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I orchestrate 4D parallelism for LLM pretraining across multiple GPUs?▼

You can orchestrate 4D parallelism for LLM pretraining by using TorchTitan to composable scale FSDP2, tensor parallelism, pipeline parallelism, and context parallelism across multi-GPU clusters.

Can I use Float8 training on H100 clusters to speed up large-scale LLM pretraining?▼

Yes, you can use Float8 training on H100 clusters to accelerate large-scale LLM pretraining, utilizing selective filtering and compile-stage acceleration for optimized speedups.

How do I launch distributed LLM pretraining jobs on 8 to 512+ GPU clusters?▼

You can launch distributed LLM pretraining jobs on 8 to 512+ GPU clusters using TorchTitan's multi-node workflows, which support ready configs and SLURM-based launches for large-scale execution.

Does TorchTitan support checkpoint interoperability with HuggingFace for pretraining workflows?▼

Yes, TorchTitan supports checkpoint interoperability with HuggingFace and torchtune, allowing you to save, resume, and convert distributed checkpoints seamlessly during LLM pretraining.

What models are supported for PyTorch-native distributed pretraining?▼

PyTorch-native distributed pretraining supports Llama 3.1, DeepSeek V3, and custom models, enabling you to optimize large-scale training workflows across multi-node GPU clusters.