Wentao-Xu

pytorch_geometric

Graph Neural Network Library for PyTorch

qlib

Qlib is an AI-oriented quantitative investment platform that aims to realize the potential, empower research, and create value using AI technologies in quantitative investment, from exploring ideas to implementing productions. Qlib supports diverse machine learning modeling paradigms. including supervised learning, market dynamics modeling, and RL.

Conference-Acceptance-Rate

Acceptance rates for the major AI conferences

SupContrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)

torchscale

Foundation Architecture for (M)LLMs

pyGAT

Pytorch implementation of the Graph Attention Network model by Veličković et. al (2017, https://arxiv.org/abs/1710.10903)

benchmarking-gnns

Repository for benchmarking graph neural networks

ogb

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

HAN

Heterogeneous Graph Neural Network

Graph_Transformer_Networks

Graph Transformer Networks (Authors' PyTorch implementation for the NeurIPS 19 paper)

KGE

Some papers on Knowledge Graph Embedding(KGE)

MTGNN

pyHGT

Code for "Heterogeneous Graph Transformer" (WWW'20), which is based on pytorch_geometric

multivariate-time-series-data

DGI

Deep Graph Infomax (https://arxiv.org/abs/1809.10341)

SCINet

The GitHub repository for the paper: “Time Series is a Special Sequence: Forecasting with Sample Convolution and Interaction“. (NeurIPS 2022)

MAGNN

Metapath Aggregated Graph Neural Network for Heterogeneous Graph Embedding

transformer-tensorflow

TensorFlow implementation of 'Attention Is All You Need (2017. 6)'

CorrectAndSmooth

[ICLR 2021] Combining Label Propagation and Simple Models Out-performs Graph Neural Networks (https://arxiv.org/abs/2010.13993)

kbc

Tools for state of the art Knowledge Base Completion.

HIST

The source code and data of the paper "HIST: A Graph-based Framework for Stock Trend Forecasting via Mining Concept-Oriented Shared Information".

SuperGAT

[ICLR 2021] How to Find Your Friendly Neighborhood: Graph Attention Design with Self-Supervision

PyTorch-Distributed-Training

Example of PyTorch DistributedDataParallel

KGE-DURA

The code of paper Duality-Induced Regularizer for Tensor Factorization Based Knowledge Graph Completion. Zhanqiu Zhang, Jianyu Cai, Jie Wang. NeurIPS 2020.

SEEK

Source code for the ACL 2020 paper "SEEK: Segmented Embedding of Knowledge Graphs".

HGNN-AC

Source code of "WWW21 - Heterogeneous Graph Neural Network via Attribute Completion"

IGMTF

The source code and data of the paper "Instance-wise Graph-based Framework for Multivariate Time Series Forecasting".

KGE-CL

Source code for the COLING 2022 paper "KGE-CL: Contrastive Learning of Tensor Decomposition Based Knowledge Graph Embeddings".

HGCN

SHGNN

The source code of the paper "SHGNN: Structure-Aware Heterogeneous Graph Neural Network"