Train large language models across many GPUs.
Guides distributed language-model pretraining with TorchTitan, including parallelism configuration, Float8 training, and checkpoints.
When to use it
Use when pretraining language models from scratch and needing PyTorch-native, composable distributed parallelism.
Give it a model-training project; it configures and launches distributed pretraining with monitoring and checkpoints.
What you provide
Declares torch version 2.6.0 or newer as a dependency.
Declares torchtitan version 0.2.0 or newer as a dependency and provides installation instructions.
This skill
Hugging Face
Downloads a Hugging Face tokenizer
GitHub
Checks torchao performance benchmarks