← all repositories

pytorch/torchtitan

A PyTorch-native training platform for large-scale generative AI model development.

5.4k stars Python ML FrameworksLanguage Models
torchtitan
Velocity · 7d
+6.0
★ / day
Trend
steady
star history

Torchtitan is a comprehensive training platform built natively in PyTorch for training generative AI models at scale. It provides distributed training capabilities, supports multiple model architectures, and includes optimizations for GPU training workflows. The platform is actively maintained by the PyTorch team and has been published at ICLR 2025.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.