Cyue0316

STG-NRDE

"Graph Neural Rough Differential Equations for Traffic Forecasting", ACM TIST

Draw_Trajectory

The repository is used to visualize trajectories in the domain of pedestrian trajectory prediction

LaneSegNet

[ICLR 2024] Map Learning with Lane Segment for Autonomous Driving

FusionAD

Official Code Release of "FusionAD"

SGCN

Code for "SGCN:Sparse Graph Convolution Network for Pedestrian Trajectory Prediction" CVPR 2021

LMTrajectory

Official Code for "Can Language Beat Numerical Regression? Language-Based Multimodal Trajectory Prediction (CVPR 2024)"

PeFAD

ABC

pytorch implementation of ABC : Auxiliary Balanced Classifier for Class-imbalanced Semi-supervised Learning

LOCAL

Prompt-align

[ICCV 2023] Prompt-aligned Gradient for Prompt Tuning

IntegratedGradientsPytorch

Integrated gradients attribution method implemented in PyTorch

GPGraph

Official Code for "Learning Pedestrian Group Representations for Multi-modal Trajectory Prediction (ECCV 2022)"

Social-STGCNN

Code for "Social-STGCNN: A Social Spatio-Temporal Graph Convolutional Neural Network for Human Trajectory Prediction" CVPR 2020

ControllableTimeSeriesGeneration

efficient-kan

An efficient pure-PyTorch implementation of Kolmogorov-Arnold Network (KAN).

gde

Neural Graph Differential Equations (Neural GDEs)

diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Neural-Manifold-Ordinary-Differential-Equations

[NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)

Neural-Modules-for-Differential-Algebraic-Equations

NeuralABM

Neural parameter calibration for multi-agent models. Uses neural networks to estimate marginal densities on parameters and networks

DLRT-Net

Dynamical low rank training for neural networks

NeuralCDE

Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)

imbalanced-regression

[ICML 2021, Long Talk] Delving into Deep Imbalanced Regression

saliency

Framework-agnostic implementation for state-of-the-art saliency methods (XRAI, BlurIG, SmoothGrad, and more).

CDMAD

Code of the CDMAD: Class-Distribution-Mismatch-Aware Debiasing for Class-Imbalanced Semi-Supervised Learning (2024 CVPR accepted paper)

Propagation-SCM

Demo code for "Causal Discovery in Linear Structural Causal Models with Deterministic Relations" (Proc. CLeaR 2022)

variational-backdoor-adjustment

TECDI

Data and codes of the paper "Causal Discovery in Temporal Domain from Interventional Data".

TV_DBN

SparseRC

Learning DAGs from Data with Few Root Causes