pytorch_geometric

PyTorch Geometric（PyG）是一个基于PyTorch构建的图神经网络（GNN）开源库，用于高效实现和训练面向图结构数据的深度学习模型。它提供统一的张量接口、丰富的GNN模型实现（如GCN、GAT、GraphSAGE）、支持小图批量加载与单一大图训练、多GPU并行、torch.compile优化及DataPipe数据流水线，并内置大量标准基准数据集（如OGB）和针对图、点云、3D网格的预处理变换。适用于科研实验、图表示学习、社交网络分析、分子建模、推荐系统等需处理非欧几里得结构数据的场景。