GraphLearn-for-PyTorch首頁、文檔和下載- 基於PyTorch 的圖學習庫- 程式开发

GraphLearn-for-PyTorch (GLT) 是一個基於PyTorch的圖學習庫,目的是使分佈式GNN訓練和推理變得簡單和高效。

GLT利用GPU加速圖採樣,並利用UVA來減少頂點和邊特徵的轉換和復制。對於大規模圖,它支持通過高效的分佈式採樣和特徵查找在多個GPU或多台機器上進行分佈式訓練。此外,GLT提供了靈活的分佈式訓練部署,以滿足不同的需求。

GTL 架構如下:

#GraphLearnforPyTorch首頁文檔和下載 #基於PyTorch #的圖學習庫 #程式开发

發佈留言

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *