PyTorch

PyTorch 2.14 Release Blog

We are excited to announce the release of PyTorch® 2.14 (release notes)! The PyTorch 2.14 release features the following changes: NVGEMM brings CuTeDSL-generated CUTLASS kernels to Inductor, with epilogue fusion,...

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....