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…