Cheng Wan's repositories
awesome-gnn-systems
A list of awesome GNN systems.
cogdl
CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)
gpustat
📊 A simple command-line utility for querying and monitoring GPU status
marius
Large scale graph learning on a single machine.
PipeGCN
[ICLR 2022] "PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communication" by Cheng Wan, Youjie Li, Cameron R. Wolfe, Anastasios Kyrillidis, Nam Sung Kim, Yingyan Lin
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
pytorch_geometric
Geometric Deep Learning Extension Library for PyTorch
ViT-pytorch
Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.