roggirg

madrona

gpudrive

GPU-acceleration of Nocturne via Madrona

FJMP

[CVPR 2023] Official repository for FJMP: Factorized Joint Multi-Agent Motion Prediction

trajdata

A unified interface to many trajectory forecasting datasets.

traffic-behavior-simulation

scenarionet

ScenarioNet: Scalable Traffic Scenario Management System for Autonomous Driving

DirectBehaviorSpecification

Code to reproduce the Arena environment experiments from Direct Behavior Specification via Constrained Reinforcement Learning.

nocturne

A data-driven, fast driving simulator for multi-agent coordination under partial observability.

envpool

C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.

speedrun

Research code need not be ugly.

yahoo_fin

Scrape stock price history from new (Spring 2017) Yahoo Finance layout

social-driving

Design multi-agent environments and simple reward functions such that social driving behavior emerges

SMARTS

Scalable Multi-Agent RL Training School for Autonomous Driving

set_transformer

Pytorch implementation of set transformer

dreamer-pytorch

Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.

ml-multiple-futures-prediction

dreamer-pytorch

pytorch-implementation of Dreamer (Model-based Image RL Algorithm)

HighwayEnv

A minimalist environment for decision-making in autonomous driving

probabilistic_robotics

solution of exercises of the book "probabilistic robotics"

Reinforcement-Learning-2nd-Edition-by-Sutton-Exercise-Solutions

Solutions of Reinforcement Learning, An Introduction

JAAD

Annotation data for JAAD (Joint Attention in Autonomous Driving) Dataset

waymo-open-dataset-viewer

A WebGL viewer for pointclouds of the Waymo Open Dataset

slac

Stochastic Latent Actor-Critic: Deep Reinforcement Learning with a Latent Variable Model

autograd

Efficiently computes derivatives of NumPy code.

pytorch-a2c-ppo-acktr-gail

PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).

keras_to_tensorflow

General code to convert a trained keras model into an inference tensorflow model

AndroidTensorFlowMachineLearningExample

Android TensorFlow MachineLearning Example (Building TensorFlow for Android)