Tensors and Dynamic neural networks in Python with strong GPU acceleration

OpenMMLab Detection Toolbox and Benchmark

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

Uniform Manifold Approximation and Projection

A Python Package to Tackle the Curse of Imbalanced Datasets in Machine Learning

Denoising Diffusion Probabilistic Models

A unified, comprehensive and efficient recommendation library

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

neuralnetworksanddeeplearning.com integrated scripts for Python 3.5.2 and Theano with CUDA support

CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)

Pytorch implementation of Diffusion Models (https://arxiv.org/pdf/2006.11239.pdf)

Samples for users of the Yelp Academic Dataset

Code for CRATE (Coding RAte reduction TransformEr).

Next RecSys Library

This is a repository of public data sources for Recommender Systems (RS).

The PyTorch implementation of LightGCN

Riemannian Adaptive Optimization Methods with pytorch optim

Source code for end-to-end dialogue model from the MultiWOZ paper (Budzianowski et al. 2018, EMNLP)

Deep learning software to decode EEG, ECG or MEG signals

Neural Graph Collaborative Filtering, SIGIR2019

SASRec: Self-Attentive Sequential Recommendation

Hyperbolic Graph Convolutional Networks in PyTorch.

Implementation of “DreamDiffusion: Generating High-Quality Images from Brain EEG Signals”

Dialogue model that produces empathetic responses when trained on the EmpatheticDialogues dataset.

The repo for Tsinghua summer course: Interdisciplinary Seminar on Big Models

Calculate the distance between 2 points on Earth

PyTorch Implementation for Neural Graph Collaborative Filtering

Data and codes for ACL 2021 paper: Towards Emotional Support Dialog Systems