Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]

(Official) PyTorch implementation for Efficient Episodic Memory Utilization of Cooperative Multi-Agent Reinforcement Learning (EMU) (ICLR 2024, Oral)

Model-Based Transfer Learning for Contextual Reinforcement Learning (NeurIPS 2024)

[arXiv 2024] "HEIGHT: Heterogeneous Interaction Graph Transformer for Robot Navigation in Crowded and Constrained Environments"

Large Driving Models

A collection of 100+ pre-trained RL agents using Stable Baselines, training and hyperparameter optimization included.

Information-Driven Design of Imaging Systems

Heterogeneous Pre-trained Transformer (HPT) as Scalable Policy Learner.

VBD: Versatile Behavior Diffusion for Generalized Traffic Agent Simulation

[ECCV 2024] Asynchronous Large Language Model Enhanced Planner for Autonomous Driving

Code for the manim-generated scenes used in 3blue1brown videos

Diffusion Explainer: Visual Explanation for Text-to-image Stable Diffusion

Transformer Explained Visually: Learn How LLM Transformer Models Work with Interactive Visualization

"OpenCity: Open Spatio-Temporal Foundation Models for Traffic Prediction"

Hunyuan-DiT : A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understanding

Implicit Game-Theoretic MPC

Truncated Diffusion Model for Real-Time End-to-End Autonomous Driving

code for ROMANCE

Code for paper "Multi-agent Continual Coordination via Progressive Task Contextualization"

A package for statistically rigorous scientific discovery using machine learning. Implements prediction-powered inference.

🔥 TeaMs-RL: Teaching LLMs to Generate Better Instruction Datasets via Reinforcement Learning

"Safety-Critical Model Predictive Control with Discrete-Time Control Barrier Function" by J. Zeng, B. Zhang and K. Sreenath https://arxiv.org/abs/2007.11718

Safe Multi-Agent Reinforcement Learning to Make decisions in Autonomous Driving.

Self-playing Adversarial Language Game Enhances LLM Reasoning, NeurIPS 2024

NAGphormer: A Tokenized Graph Transformer for Node Classification in Large Graphs

A comprehensive list of papers for the definition of World Models and using World Models for General Video Generation, Embodied AI, and Autonomous Driving, including papers, codes, and related webs…

Implementation of DreamerV3 in Pytorch