GraphLearn-for-PyTorch基于 PyTorch 的图学习库

联合创作 · 2023-09-26 06:59

GraphLearn-for-PyTorch (GLT) 是一个基于PyTorch的图学习库,目的是使分布式GNN训练和推理变得简单和高效。

GLT利用GPU加速图采样,并利用UVA来减少顶点和边特征的转换和复制。对于大规模图,它支持通过高效的分布式采样和特征查找在多个GPU或多台机器上进行分布式训练。此外,GLT提供了灵活的分布式训练部署,以满足不同的需求。

GTL 架构如下:

浏览 3
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报