pytorch_geometric

A PyTorch-based library for graph neural networks offering GNN models, data loaders, and benchmark datasets for researchers and engineers working with graphs, meshes, and point clouds.