ICLRMar, 2022

针对原子模拟的十亿参数图神经网络训练

TL;DR本文提出的图并行算法可在多个 GPU 上分布输入图,从而实现训练数十亿个参数的大型 GNN, 该方法在大规模开放铂族催化剂的数据集上取得了新的最先进结果。