YukeWang96/TC-GNN_ATC23
Artifact for USENIX ATC'23: TC-GNN: Bridging Sparse GNN Computation and Dense Tensor Cores on GPUs.
This project provides an optimized way to perform Graph Neural Network (GNN) computations on NVIDIA GPUs. It takes your GNN models and data, processing them more efficiently by leveraging specialized hardware on modern GPUs. This tool is designed for deep learning researchers and engineers who work with GNNs and need to accelerate their model training or inference.
No commits in the last 6 months.
Use this if you are a deep learning practitioner working with Graph Neural Networks and want to significantly speed up your computations on NVIDIA Ampere-generation GPUs.
Not ideal if you are not working with Graph Neural Networks or do not have access to an NVIDIA GPU with compute capability 8.0 or higher.
Stars
55
Forks
16
Language
Python
License
—
Category
Last pushed
Oct 16, 2023
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/YukeWang96/TC-GNN_ATC23"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
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.