dgl

收录理由

如果你的机器学习工作经常需要处理图结构数据，DGL 能让你直接在上面训练神经网络，而不必自己从头实现消息传递机制。它提供的图对象可以放在 CPU 或 GPU 上，把结构信息与节点、边的特征捆绑在一起，同时带有高效且可定制的消息传递原语，让 GNN 代码保持清晰易读。这个库与具体框架无关，周边应用可以用 PyTorch 或 TensorFlow 编写，图模型作为其中一个组件接入即可。项目还收录了大量覆盖多种 GNN 模型家族的示例实现，DGL-Go 则提供了命令行界面，让你无需编写样板代码就能训练和研究前沿模型。对于要搭建真实图深度学习系统、而非只做单个玩具模型的人来说，这是一个成熟且文档完善的基础。