wuhaixu2016

minGPT

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Awesome-Diffusion-Models

A collection of resources and papers on Diffusion Models

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

Informer2020

The GitHub repository for the paper "Informer" accepted by AAAI 2021.

Time-Series-Library

A Library for Advanced Deep Time Series Models.

decision-transformer

Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.

Autoformer

About Code release for "Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecasting" (NeurIPS 2021), https://arxiv.org/abs/2106.13008

pytorch_scatter

PyTorch Extension Library of Optimized Scatter Operations

tods

TODS: An Automated Time-series Outlier Detection System

EfficientZero

Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.

Anomaly-Transformer

About Code release for "Anomaly Transformer: Time Series Anomaly Detection with Association Discrepancy" (ICLR 2022 Spotlight), https://openreview.net/forum?id=LzQQ89U1qm_

TimesNet

About Code release for "TimesNet: Temporal 2D-Variation Modeling for General Time Series Analysis" (ICLR 2023), https://openreview.net/pdf?id=ju_Uqw384Oq

Nonstationary_Transformers

Code release for "Non-stationary Transformers: Exploring the Stationarity in Time Series Forecasting" (NeurIPS 2022), https://arxiv.org/abs/2205.14415

predrnn-pytorch

Official implementation for NIPS'17 paper: PredRNN: Recurrent Neural Networks for Predictive Learning Using Spatiotemporal LSTMs.

GFNet

[NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification

Sophia

Effortless plugin and play Optimizer to cut model training costs by 50%. New optimizer that is 2x faster than Adam on LLMs.

depyf

depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile.

Flowformer

About Code release for "Flowformer: Linearizing Transformers with Conservation Flows" (ICML 2022), https://arxiv.org/pdf/2202.06258.pdf

Corrformer

About code release of "Interpretable Weather Forecasting for Worldwide Stations with a Unified Deep Model", Nature Machine Intelligence, 2023. https://www.nature.com/articles/s42256-023-00667-9

A-Roadmap-for-Transfer-Learning

PyFleX

Customized Python APIs for NVIDIA FleX

Latent-Spectral-Models

About Code Release for "Solving High-Dimensional PDEs with Latent Spectral Models" (ICML 2023), https://arxiv.org/abs/2301.12664

MotionRNN

About Code release for "MotionRNN: A Flexible Model for Video Prediction with Spacetime-Varying Motions" (CVPR 2021) https://arxiv.org/abs/2103.02243

ContextWM

Code release for "Pre-training Contextualized World Models with In-the-wild Videos for Reinforcement Learning" (NeurIPS 2023), https://arxiv.org/abs/2305.18499

SimpleTrack

SimpleTrack: Rethinking and Improving a JDE Approach for Multi-Object Tracking

VGPL-Dynamics-Prior

[ICML 2020] Visual Grounding of Learned Physical Models (Dynamics Prior)

Iso-Dream

SPOT

Code release for "Supported Policy Optimization for Offline Reinforcement Learning" (NeurIPS 2022), https://arxiv.org/abs/2202.06239

ModeRNN