2 papers
cs.LG2026
A Dependency-Aware Multi-GPU Training Framework for Temporal Graph Neural Networks
Yang Zhao, Yue Dai, Liang Qiao +8
Temporal graph neural networks (TGNNs) achieve superior performance in many dynamic graph learning tasks. These models use node state vectors (i.e., memories) to capture and record…
cs.CV2026
TERRA: A Hierarchical Parallel Training and Memory Orchestration Framework for High-Resolution AI-based Earth Modeling
Ruohan Wu, Ziqi Zhu, Yang Zhao +6
Training high-resolution AI-based Earth forecasting models is memory-intensive. Window-based Swin Transformers reduce the quadratic cost of global attention, but existing distribut…