← all repositories

pyg-team/pytorch_geometric

A PyTorch-based library for deep learning on graph-structured data, point clouds, and 3D meshes.

23.8k stars Python ML Frameworks
pytorch_geometric
Velocity · 7d
+7.5
★ / day
Trend
steady
star history

PyTorch Geometric provides implementations of graph neural network layers, benchmark datasets, mini-batch loaders, and training utilities for working with structured data such as graphs, meshes, and point clouds. It supports multi-GPU training, torch.compile optimization, and implements methods from published geometric deep learning papers.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.