Temporal Graph Convolutional Network for Urban Traffic Flow Prediction Method

Benchmark datasets, data loaders, and evaluators for graph machine learning

[ICML 2021] "Do We Actually Need Dense Over-Parameterization? In-Time Over-Parameterization in Sparse Training" by Shiwei Liu, Lu Yin, Decebal Constantin Mocanu, Mykola Pechenizkiy

[Neurips 2021] Sparse Training via Boosting Pruning Plasticity with Neuroregeneration

Graphormer is a deep learning package that allows researchers and developers to train custom models for molecule modeling tasks. It aims to accelerate the research and application in AI for molecul…

[Preprint] "Bag of Tricks for Training Deeper Graph Neural Networks A Comprehensive Benchmark Study" by Tianlong Chen*, Kaixiong Zhou*, Keyu Duan, Wenqing Zheng, Peihao Wang, Xia Hu, Zhangyang Wang

[ICML 2021] "A Unified Lottery Tickets Hypothesis for Graph Neural Networks", Tianlong Chen*, Yongduo Sui*, Xuxi Chen, Aston Zhang, Zhangyang Wang

PyTorch implementation of Masked Autoencoder

Official implementation for "Pruning Randomly Initialized Neural Networks with Iterative Randomization"

Author's PyTorch implementation of TD3 for OpenAI gym tasks

Robust Reinforcement Learning with the Alternating Training of Learned Adversaries (ATLA) framework

A pytorch adversarial library for attack and defense methods on images and graphs

Simple PyTorch implementations of U-Net/FullyConvNet (FCN) for image segmentation

Pytorch Implementation of Twin Delayed Deep Deterministic Policy Gradients for Continuous Control

Pytorch ImageNet training codes with various tricks, lr schedulers, distributed training, mixed precision training, DALI dataloader etc.

PyTorch implementation of PPO algorithm