Lei Zhang

Lei Zhang is an MTS Software Development Engineer on the Training at Scale team, driving scheduler efficiency and training stability for cloud-native training platforms at scale.

Posts by Lei Zhang

February 08, 2026

Resilient Large-Scale Training: Integrating TorchFT with TorchTitan on AMD GPUs

Achieve resilient, checkpoint-less distributed training on AMD GPUs by integrating TorchFT with TorchTitan on Primus-SaFE.

https://rocm.blogs.amd.com/artificial-intelligence/primus-torchft/README.html