Showing cs.LGShow all
2 papers · 1 filter
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.LG2024
Pruner: A Draft-then-Verify Exploration Mechanism to Accelerate Tensor Program Tuning
Liang Qiao, Jun Shi, Xiaoyu Hao +10
Tensor program tuning is essential for the efficient deployment of deep neural networks. Search-based approaches have demonstrated scalability and effectiveness in automatically fi…