Optimizing Dynamic Neural Networks with Brainstorm. - Weihao Cui; Zhenhua Han; Lingji Ouyang; Yichuan Wang 002; Ningxin Zheng; Lingxiao Ma; Yuqing Yang 001; Fan Yang 024; Jilong Xue; Lili Qiu; Lidong Zhou; Quan Chen 002; et al. (OSDI, 2023)

MuseGNN: Forming Scalable, Convergent GNN Layers that Minimize a Sampling-Based Energy. - Haitian Jiang; Renjie Liu 001; Zengfeng Huang; Yichuan Wang 002; Xiao Yan 002; Zhenkun Cai; Minjie Wang; David Wipf (ICLR, 2025)

DiskGNN: Bridging I/O Efficiency and Model Accuracy for Out-of-Core GNN Training. - Renjie Liu 001; Yichuan Wang 002; Xiao Yan 002; Haitian Jiang; Zhenkun Cai; Minjie Wang; Bo Tang 016; Jinyang Li 001 (Proc. ACM Manag. Data, 2025)