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.