An MLP-based weather model called WeatherMixer trains efficiently with a new Jigsaw parallelization scheme that shards data and model across GPUs, reaching 11 PFLOPs on 256 GPUs with 72% scaling efficiency.
Title resolution pending
1 Pith paper cite this work. Polarity classification is still indexing.
1
Pith paper citing it
fields
cs.LG 1years
2025 1verdicts
CONDITIONAL 1representative citing papers
citing papers explorer
-
Jigsaw: Training Multi-Billion-Parameter AI Weather Models with Optimized Model Parallelism
An MLP-based weather model called WeatherMixer trains efficiently with a new Jigsaw parallelization scheme that shards data and model across GPUs, reaching 11 PFLOPs on 256 GPUs with 72% scaling efficiency.