pytorch_geometric

收录理由

当数据本身是图结构——社交网络、分子、引文网络或三维点云——PyTorch Geometric 往往是团队训练神经网络时的首选。它把论文里发表的多种图神经网络层和模型都实现好了，你不需要从零写消息传递代码，直接挑一个已知架构起步就行。工程上的细节也考虑得很周到：小批量加载器既能处理大量小图，也能应对单个超大图；多 GPU 训练和 torch.compile 都支持。它还自带不少常用基准数据集，以及针对图、网格和点云的变换工具，做研究实验或搭生产管线都省事。如果你本来就用 PyTorch，这套 API 上手很自然，文档和示例笔记本能让你很快跑通第一个 GNN。