Lightlines
By orchestra-research

distributed-llm-pretraining-torchtitan

orchestra-research898 installs

Train large language models across many GPUs.

What it does

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.

How to use it

Give it a model-training project; it configures and launches distributed pretraining with monitoring and checkpoints.

What you provide

  • An existing project

Uses

Latest review

What you need · 8

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.


Access · 6

This skill

Hugging Face

Read

Downloads a Hugging Face tokenizer

GitHub

Read

Checks torchao performance benchmarks