We are pleased to introduce you graphlearn-for-pytorch (https://github.com/alibaba/graphlearn-for-pytorch), an open-source distributed graph neural network library based on PyTorch and compatible with PyG. Our library is designed to make it easy for developers to build and train large-scale graph models in a distributed environment. With graphlearn-for-pytorch, you can leverage GPUs to accelerate graph sampling and utilize UVA to reduce the overheads of feature collection. Following a scalable design, graphlearn-for-pytorch supports training GNN models on multiple GPUs or even multiple machines. We have added some examples to demonstrate how to train PyG models in the distributed setting.

You are welcome to try it out and give us your feedback, tell us your feature requests and report bugs!