Open to career opportunities and collaborations. Learn more .

TorchTitan

A PyTorch-native platform for generative model pretraining and distributed optimization.

Overview

TorchTitan provides a production-ready PyTorch-native solution for LLM pretraining, offering multi-dimensional parallelism and practical training recipes.

Key features

  • FSDP2 and various parallelism strategies.
  • Training recipes and deployment guidance.

Use cases

  • Pretraining large generative models and distributed experiments.

Technical highlights

  • Close integration with PyTorch distributed primitives and tooling.
TorchTitan
Resource Info
🖥️ ML Platform 🌱 Open Source