PyG: pytorch_geometric: https://github.com/pyg-team/pytorch_geometric