awslabs/graphstorm
Enterprise graph machine learning framework for billion-scale graphs for ML scientists and data scientists.
GraphStorm helps ML scientists and data scientists build and deploy machine learning models on extremely large datasets structured as graphs, like social networks or citation networks. It takes your raw graph data (nodes and edges) and outputs trained models that can classify items or predict connections within your data. This is for professionals who need to work with graphs containing billions of data points.
450 stars.
Use this if you need to train and deploy advanced graph machine learning models on massive datasets without extensive coding or managing complex distributed infrastructure.
Not ideal if you are working with small datasets or simple graph analysis tasks that don't require enterprise-scale model training.
Stars
450
Forks
71
Language
Python
License
Apache-2.0
Category
Last pushed
Jan 08, 2026
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/awslabs/graphstorm"
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.