PyTorch

FP8 Training on AMD GPUs with TorchTitan and TorchAO: Upstreaming Performance Improvements

At the PyTorch Conference 2025, we demonstrated linear scaling beyond 1,000 GPUs on AMD Instinct clusters using Primus-Turbo, an AMD optimization library for training frameworks such as TorchTitan. We have since upstreamed those AMD optimizations so TorchTitan supports AMD Instinct(™) GPUs directly, with competitive FP8 performance out of the box. All contributions mentioned have been…

PyTorch 2.13 Release Blog

We are excited to announce the release of PyTorch® 2.13 (release notes)! The PyTorch 2.13 release features the following changes:   FlexAttention lands on Apple Silicon (MPS), with up to...

Bringing PyTorch Monarch to AMD GPUs: Single-Controller Distributed Training on ROCm

Training state-of-the-art large language models (LLMs) with billions of parameters requires distributed training across hundreds or thousands of GPUs. At this scale, hardware failures are not exceptional events—they are expected....

PyTorch 2.12 Release Blog

We are excited to announce the release of PyTorch® 2.12 (release notes)!   The PyTorch 2.12 release features the following changes:  Batched linalg.eigh on CUDA is up to 100x faster due...