[NeurIPS 2024] SMART: Scalable Multi-agent Real-time Motion Generation via Next-token Prediction

TrafficBots: Towards World Models for Autonomous Driving Simulation and Motion Prediction. ICRA 2023. You may also want to check out the updated version: https://github.com/zhejz/TrafficBotsV1.5

Goal Driven Multi-Modal Motion Prediction Network

A library for differentiable robotics on manifolds.

TrafficBots: Towards World Models for Autonomous Driving Simulation and Motion Prediction. ICRA 2023. Code is now available at https://github.com/zhejz/TrafficBots

The devkit of the nuPlan dataset.

MTR: Motion Transformer with Global Intention Localization and Local Movement Refinement, NeurIPS 2022.

This repository provides the code used to implement the framework to provide deep learning models with total uncertainty estimates as described in "A General Framework for Uncertainty Estimation in…

A Unofficial Pytorch Implementation of TNT: Target-driveN Trajectory Prediction

Graph Neural Network Library for PyTorch

Personal Website

PyTorch implementations of deep reinforcement learning algorithms and environments

Rainbow is all you need! A step-by-step tutorial from DQN to Rainbow

TF-Agents: A reliable, scalable and easy to use TensorFlow library for Contextual Bandits and Reinforcement Learning.

Keras Implementation of popular Deep RL Algorithms (A3C, DDQN, DDPG, Dueling DDQN)

Open-source simulator for autonomous driving research.