ShardMeter: Sharded and Geo-Distributed Training Without the Guesswork
Not on the map. Why?
Abstract (arXiv)
Training large-scale AI models often outgrows a single data center, demanding sharded, multi-cluster, and decentralized training. However, the huge space of resource allocations makes exhaustive benchmarking and manual tuning impractical, while performance depends on tightly coupled factors like model size, GPU memory, batch size, bandwidth, and sharding strategy. We introduce ShardMeter, a lightweight analytical performance model that predicts the end-to-end runtime of transformer-based workloads across arbitrary sharded, distributed, and even decentralized training. Given a model's characteristics and a target hardware topology, ShardMeter estimates per-GPU and per-island throughput, training cost, total wall-clock time, and identifies performance bottlenecks. Our analysis reveals diminishing-return regimes as island size increases, quantifies transitions between compute- and communication-bound scaling, evaluates hyperparameter trade-offs, and models cost-throughput for large-scale decentralized training. ShardMeter exposes these insights to quickly explore the configuration space, choose near-optimal deployment plans, and avoid costly trial and error.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 24 of its 32 references that are in the map. First posted on arXiv 24 Aug 2026.
Builds on
24 of 32 references.
References · 24
- 2017Attention Is All You Needcited by 60,361
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2023A Survey of Large Language Modelscited by 2,576
- 2024Gemma: Open Models Based on Gemini Research and Technologycited by 916
- 2024Yi: Open Foundation Models by 01.AIcited by 719
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 660
- 2023Textbooks Are All You Need II: phi-1.5 technical reportcited by 565
- 2018An Empirical Model of Large-Batch Trainingcited by 281
12 more
- 2023DiLoCo: Distributed Low-Communication Training of Language Modelscited by 121
- 2024TorchTitan: One-stop PyTorch native solution for production ready LLM pre-trainingcited by 85
- 2024Multimodal Alignment and Fusion: A Surveycited by 74
- 2020Pollux: Co-adaptive Cluster Scheduling for Goodput-Optimized Deep Learningcited by 72
- 2018Predicting the Computational Cost of Deep Learning Modelscited by 49
- 2021A Runtime-Based Computational Performance Predictor for Deep Neural Network Trainingcited by 29
- 2025Streaming DiLoCo with overlapping communication: Towards a Distributed Free Lunchcited by 28
- 2024INTELLECT-1 Technical Reportcited by 21
- 2025Sailor: Automating Distributed Training over Dynamic, Heterogeneous, and Geo-distributed Clusterscited by 13
- 2025Hermes 4 Technical Reportcited by 8
- 2021An Oracle for Guiding Large-Scale Model/Hybrid Parallel Training of Convolutional Neural Networkscited by 2
- 2024PreNeT: Leveraging Computational Features to Predict Deep Neural Network Training Timecited by 1
Built on by
Not cited yet.