alibaba/graphlearn-for-pytorch
A GPU-accelerated graph learning library for PyTorch, facilitating the scaling of GNN training and inference.
This is a GPU-accelerated library designed to help developers efficiently train and deploy Graph Neural Networks (GNNs) on very large datasets. It takes raw graph data and features, enabling faster model training and inference by leveraging GPU power. This tool is for machine learning engineers and researchers who build and scale GNN applications.
146 stars. No commits in the last 6 months.
Use this if you are a machine learning engineer working with PyTorch and need to train GNN models on large-scale graphs, especially when you have access to GPUs and want to optimize performance.
Not ideal if you are not a developer, do not use PyTorch, or are working with small graphs where performance is not a bottleneck.
Stars
146
Forks
46
Language
Python
License
Apache-2.0
Category
Last pushed
Sep 23, 2025
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/alibaba/graphlearn-for-pytorch"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
pyg-team/pytorch_geometric
Graph Neural Network Library for PyTorch
a-r-j/graphein
Protein Graph Library
raamana/graynet
Subject-wise networks from structural MRI, both vertex- and voxel-wise features (thickness, GM...
pykale/pykale
Knowledge-Aware machine LEarning (KALE): accessible machine learning from multiple sources for...
dmlc/dgl
Python package built to ease deep learning on graph, on top of existing DL frameworks.