Monte Carlo Tree Search with Reinforcement Learning for Motion Planning

Official release of InternLM2.5 base and chat models. 1M context support

🎨 ML Visuals contains figures and templates which you can reuse and customize to improve your scientific writing.

Transformer implementation with PyTorch for remaining useful life prediction on turbofan engine with NASA CMAPSS data set. Inspired by Mo, Y., Wu, Q., Li, X., & Huang, B. (2021). Remaining useful l…

[ECCV 2024] DriveDreamer: Towards Real-world-driven World Models for Autonomous Driving

[CVPR 2024] A world model for autonomous driving.

[CoRL'23] Parting with Misconceptions about Learning-based Vehicle Motion Planning

PyTorch code for the paper "Model-Based Imitation Learning for Urban Driving".

Udacity Self-Driving Car Engineer Nanodegree: Quintic Polynomial Solver. & Paper 'Optimal Trajectory Generation for Dynamic Street Scenarios in a Frenet Frame'

Python sample codes for robotics algorithms.

[ICCV 2023 Oral] Game-theoretic modeling and learning of Transformer-based interactive prediction and planning

Pytorch implementation of paper "VectorNet: Encoding HD maps and Agent Dynamics from Vectorized Representation"

The devkit of the nuPlan dataset.

[CVPR 2023] Query-Centric Trajectory Prediction

Pytorch implementation of CVPR2020 paper “VectorNet: Encoding HD Maps and Agent Dynamics from Vectorized Representation”

[ICLR'22] A Unified Architecture for Predicting Multiple Agent Trajectories

[ECCV2020 Oral] Learning Lane Graph Representations for Motion Forecasting

Official Repository for "Interpretable Goal-based Prediction and Planning for Autonomous Driving" (ICRA 2021)

[CVPR 2023 Best Paper Award] Planning-oriented Autonomous Driving

Simulation of car following model

I implemented the reinforcement learning based model Upper Confidence Bound in both Python and R

[ICRA2020] Path Planning in Dynamic Environments using Generative RNNs and Monte Carlo Tree Search

Implementation of Convolutional LSTM in PyTorch.

Conformalized Quantile Regression

🌊 Online machine learning in Python

A scikit-learn-compatible module to estimate prediction intervals and control risks based on conformal predictions.