GraphLearn-for-PyTorch基于 PyTorch 的圖學(xué)習(xí)庫

GraphLearn-for-PyTorch (GLT) 是一個基于PyTorch的圖學(xué)習(xí)庫，目的是使分布式GNN訓(xùn)練和推理變得簡單和高效。

GLT利用GPU加速圖采樣，并利用UVA來減少頂點和邊特征的轉(zhuǎn)換和復(fù)制。對于大規(guī)模圖，它支持通過高效的分布式采樣和特征查找在多個GPU或多臺機器上進行分布式訓(xùn)練。此外，GLT提供了靈活的分布式訓(xùn)練部署，以滿足不同的需求。

GTL 架構(gòu)如下：