ElasWave: An Elastic-Native System for Scalable Hybrid-Parallel Training
Abstract
Large-scale LLM pretraining now runs across -- accelerators, making failures routine and elasticity mandatory. We posit that an elastic-native training system must jointly deliver (i) parameter consistency, (ii) low mean time to recovery (MTTR), (iii) high post-change throughput, and (iv) computation consistency. No prior system achieves all four simultaneously. To achieve these goals, we present ElasWave, which delivers per-step fault tolerance via multi-dimensional scheduling across graph, dataflow, DVFS, and RNG. ElasWave reshapes and reshards micro-batches while preserving the global batch size and gradient scale. It performs online pipeline resharding with asynchronous parameter migration and interleaves ZeRO partitions, reducing parameter recovery processes to disjoint rank-to-rank transfers. It further leverages DVFS to absorb pipeline bubbles and reshards RNG to keep computation consistency. Together, a dynamic communicator enables in-place communication group edits, while per-step in-memory snapshots support online verification and redistribution. We evaluate ElasWave on 96 NPUs and benchmark it against state-of-the-art baselines: throughput improves by over ReCycle and over TorchFT; communicator recovery completes within one second (up to faster than full/partial rebuilds); migration MTTR drops by as much as ; and convergence deviation is reduced by approximately .
Cite
@article{arxiv.2510.00606,
title = {ElasWave: An Elastic-Native System for Scalable Hybrid-Parallel Training},
author = {Xueze Kang and Guangyu Xiang and Yuxin Wang and Hao Zhang and Yuchu Fang and Yuhang Zhou and Zhenheng Tang and Youhui Lv and Eliran Maman and Mark Wasserman and Alon Zameret and Zhipeng Bian and Shushu Chen and Zhiyou Yu and Jin Wang and Xiaoyu Wu and Yang Zheng and Chen Tian and Xiaowen Chu},
journal= {arXiv preprint arXiv:2510.00606},
year = {2025}
}