code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"

MoGe: Unlocking Accurate Monocular Geometry Estimation for Open-Domain Images with Optimal Training Supervision

Learn online intrinsic rewards from LLM feedback

Teleoperation of LEAP Hand using Manus Gloves

Official Implementation for our NeurIPS 2024 paper, "Don't Look Twice: Run-Length Tokenization for Faster Video Transformers".

Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch

Emulate Nintendo Switch Controllers over Bluetooth

A Nintendo Switch Pro Controller Linux device driver (wired) in Python.

Official implementation of NeurIPS 2024 paper: "Omnigrasp: Simulated Humanoid Grasping on Diverse Objects".

Official Implementation of the ICLR 2024 spotlight paper: Universal Humanoid Motion Representations for Physics-Based Control

System identification with MuJoCo

Real-time behaviour synthesis with MuJoCo, using Predictive Control

Python inverse kinematics based on MuJoCo

A Python package that provides evaluation and visualization tools for the DexYCB dataset

[ICCV 2023] Official PyTorch implementation of the paper "InterDiff: Generating 3D Human-Object Interactions with Physics-Informed Diffusion"

[ ECCV 2024 ] MotionLCM: This repo is the official implementation of "MotionLCM: Real-time Controllable Motion Generation via Latent Consistency Model"

Puzzles for learning Triton

Schedule-Free Optimization in PyTorch

Collection of high-quality robo learning papers for bipedal robots.

Abridged 'textbooks' for future reference from the University of Toronto's Engineering Science program (Machine Intelligence option).

Parallel Q-Learning: Scaling Off-policy Reinforcement Learning under Massively Parallel Simulation

Really Fast End-to-End Jax RL Implementations

A minimal and stable PPO.

Convolutions for Sequence Modeling