Stars
InstantID: Zero-shot Identity-Preserving Generation in Seconds 🔥
DC3: A Learning Method for Optimization with Hard Constraints
[ECCV 2024] 3D World Model for Autonomous Driving
Uncertainty-aware Semantic Segmentation of LiDAR Point Clouds for Autonomous Driving
Velocity corrected Iterative Compact Ellipsoidal Transform
MTR: Motion Transformer with Global Intention Localization and Local Movement Refinement, NeurIPS 2022.
An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)
A simple tool that can generate TensorRT plugin code quickly.
Causal Inference Mixtape Sessions
PyTorch code for the paper "Model-Based Imitation Learning for Urban Driving".
Simple C++ Image Steganography tool to encrypt and hide files insde images using Least-Significant-Bit encoding.
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
A tutorial for getting started with the Deep Learning Accelerator (DLA) on NVIDIA Jetson
An automatic code generator for nonlinear model predictive control (NMPC) and the continuation/GMRES method (C/GMRES) based numerical solvers for NMPC
Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022
PyData London 2022 Tutorial
[ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
Training materials associated with NVIDIA's CUDA Training Series (www.olcf.ornl.gov/cuda-training-series/)
An implementation of the POMCP algorithm in python in the hopes of learning something new.
Algorithm and model experiments for robot motion planning. Implemented in Julia.