Stars
A minimal implementation of DeepMind's Genie world model
Official implementation of EC-Diffuser: Multi-Object Manipulation via Entity-Centric Behavior Generation
Official PyTorch implementation of "Entity-Centric Reinforcement Learning for Object Manipulation from Pixels", Haramati et al., ICLR 2024
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
This Python library makes it easy to display images and videos in a notebook.
A community-maintained Python framework for creating mathematical animations.
A small, interpretable codebase containing the re-implementation of a few "deep" NLP models in PyTorch. Colab notebooks to run with GPUs. Models: word2vec, CNNs, transformer, gpt.
Official code for the paper "Semantically Driven Sentence Fusion: Modeling and Evaluation".
The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
Mastering Blackjack - Combining Supervised Learning with Deep Reinforcement Learning
Python3 ROS Interface with the Franka Emika Panda robot arm. Includes support for OpenAI Gym based interfaces.
S-RL Toolbox: Reinforcement Learning (RL) and State Representation Learning (SRL) for Robotics
A PyTorch re-implementation of Neural Radiance Fields
Simple and easily configurable grid world environments for reinforcement learning
Information Distillation Generative Adversrial Network in PyTorch
A pytorch implementation of Paper "IntroVAE: Introspective Variational Autoencoders for Photographic Image Synthesis"
PyTorch Agent Net: reinforcement learning toolkit for pytorch
Sequence to sequence learning using TensorFlow.