PyTorch
Read post

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

PyTorch Monarch, Meta's single-controller distributed training framework, has been ported to AMD Instinct GPUs running ROCm. The port involved converting CUDA/NCCL code to HIP/RCCL using hipify_torch, handling dynamic linking differences, and adding a Rust compatibility shim to avoid platform-specific branching. Integrated with TorchTitan and TorchFT, the system enables checkpoint-less fault-tolerant training: when a GPU process crashes, healthy replicas continue training while the failed replica restarts and receives a peer checkpoint transfer from a donor replica. Validation on a 16-node SLURM cluster (128 MI300 GPUs) and a 32-node Kubernetes cluster (256 MI355 GPUs) showed stable loss convergence despite frequent injected failures, with no full job restarts required.

    #machine-learning#llm#pytorch#rocm
Jul 06•10m read time•From pytorch.org
Post cover image
Table of contents
The Challenge: Reliability at ScaleWhat is PyTorch Monarch?Porting Monarch to ROCm: Ecosystem IntegrationCase Study: Fault-Tolerant Training at ScalePerformance CharacteristicsSummary and Future DirectionsAdditional ResourcesDisclaimers
255 Impressions
PyTorch's image
PyTorch

PyTorch offers insights into deep learning, neural network modeling, and machine learning research, ...

142 Followers

•

231 Upvotes

Would you recommend this post?

Copy link
WhatsApp
Facebook
X
New Squad
  • © 2026 Daily Dev Ltd.
  • Guidelines
  • Explore
  • Tags
  • Sources
  • Squads
  • Leaderboard