ShardMeter: Sharded and Geo-Distributed Training Without the Guesswork
cs.DC, cs.AI
Submitted: 2026-08-24
Updated: 2026-08-24
Code: https://github.com/PrimeIntellect-ai/prime-rl
License: http://creativecommons.org/licenses/by-nc-nd/4.0/
The gist: Training large-scale AI models often outgrows a single data center, demanding sharded, multi-cluster, and decentralized training.
Terminology
Abstract
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.
Sources
- Attention Is All You Need
- A Survey of Large Language Models
- An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale
- Learning Transferable Visual Models From Natural Language Supervision
- Multimodal Alignment and Fusion: A Survey
- PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallel
- Hermes 4 Technical Report
- INTELLECT-1 Technical Report
- Communication-Efficient Learning of Deep Networks from Decentralized Data
- An Empirical Model of Large-Batch Training
- DiLoCo: Distributed Low-Communication Training of Language Models
- Streaming DiLoCo with overlapping communication: Towards a Distributed Free Lunch
- The Llama 3 Herd of Models
- Textbooks Are All You Need II: phi-1.5 technical report
- Gemma: Open Models Based on Gemini Research and Technology
- Yi: Open Foundation Models by 01.AI
- Predicting the Computational Cost of Deep Learning Models
Related papers
- iScheduler: Reinforcement Learning-Driven Continual Optimization for Large-Scale Resource Investment Problems
- SAMM: Sharded Automated Market Maker
- InferScale: GPU-Native KV Injection for Personalized LLM Serving
- Vigil: Accountable Liveness against Selective Silence
- Steelhead: Interleaving Partially Synchronous and Asynchronous Commit Rules on a Shared DAG
- Pushing CPU Speech Synthesis to the Wall: Extreme Inference Tuning under Serverless Architecture and Billing