RoboCasa: Large-Scale Simulation of Everyday Tasks for Generalist Robots

Fine-Tuning Vision-Language-Action Models: Optimizing Speed and Success

An example RLDS dataset builder for X-embodiment dataset conversion.

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

dataloading is my passion

Efficiently apply modification functions to RLDS/TFDS datasets.

Octo is a transformer-based robot policy trained on a diverse mix of 800k robot trajectories.

robomimic: A Modular Framework for Robot Learning from Demonstration

Instant neural graphics primitives: lightning fast NeRF and more

Clean PyTorch implementations of imitation learning algorithms

PyTorch version of Stable Baselines, improved implementations of reinforcement learning algorithms.

Unified framework for robot learning built on NVIDIA Isaac Sim

Image-to-Image Translation in PyTorch

DrQ: Data regularized Q

An open source robotics benchmark for meta- and multi-task reinforcement learning

DrQ-v2: Improved Data-Augmented Reinforcement Learning