PyTorch Geometric (PyG)新

PyTorch Geometric（PyG）是基于 PyTorch 的图神经网络库，提供了丰富的 GNN 层与消息传递算子（GCN、GAT、GINE、SchNet、GemNet 等），以及图采样器、标准数据集与异构图支持。

对物理 AI 的价值：在物理 AI 领域，它是分子、晶体与物理系统图表示学习的事实标准——支撑 GNN 原子间势、分子性质预测与拓扑感知的代理模型。