TrainMover: An Interruption-Resilient Runtime for ML Training
Not on the map. Why?
Abstract (arXiv)
Large-scale ML training jobs are frequently interrupted by hardware and software anomalies, failures, and management events. Existing solutions like checkpoint-restart or runtime reconfiguration suffer from long downtimes and degraded performance. We present TrainMover, a resilient LLM training runtime that leverages elastic and standby machines to handle interruptions with minimal downtime and zero memory overhead. To achieve these goals, TrainMover introduces three key techniques: two-phase, delta-based communication group setup; communication-free sandboxed warmup; and general standby design that enables failure recovery from any role. Our evaluation shows that TrainMover consistently achieves around 20 seconds of downtime when handling various interruptions at the 1024-GPU scale. TrainMover is projected to reduce wasted GPU hours by 55% compared to the best alternative, saving 1.4 million GPU-hours per week at the 64K-GPU scale.
Key facts. Cited by 1 paper in the map (data through 13 Oct 2026). Builds on 18 of its 49 references that are in the map. First posted on arXiv 17 Dec 2024.
Builds on
18 of 49 references.
References · 18
- 2020Language Models are Few-Shot Learnerscited by 32,320
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,072
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,979
- 2016Pointer Sentinel Mixture Modelscited by 3,047
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,851
- 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,289
- 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 626
- 2019Deep Learning Recommendation Model for Personalization and Recommendation Systemscited by 327
- 2021ZeRO-Infinity: Breaking the GPU Memory Wall for Extreme Scale Deep Learningcited by 255
- 2019NeMo: a toolkit for building AI applications using Neural Modulescited by 232
- 2024MegaScale: Scaling Large Language Model Training to More Than 10,000 GPUscited by 111
- 2021Colossal-AI: A Unified Deep Learning System For Large-Scale Parallel Trainingcited by 96
6 more
- 2022Bamboo: Making Preemptible Instances Resilient for Affordable Training of Large DNNscited by 49
- 2023Oobleck: Resilient Distributed Training of Large Models Using Pipeline Templatescited by 40
- 2024Revisiting Reliability in Large-Scale Machine Learning Research Clusterscited by 28
- 2024ByteCheckpoint: A Unified Checkpointing System for Large Foundation Model Developmentcited by 24
- 2024Parcae: Proactive, Liveput-Optimized DNN Training on Preemptible Instancescited by 17
- 2024ReCycle: Resilient Training of Large DNNs using Pipeline Adaptationcited by 14
Built on by
Citing papers per year: 2026: 1.
1 paper.