siyan-zhao / decision-stacks
Implementation of Decision Stacks: Flexible RL via Modular Generative Models
☆12Updated last year
Related projects ⓘ
Alternatives and complementary repositories for decision-stacks
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆30Updated last year
- ☆15Updated last year
- Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022☆43Updated 8 months ago
- ☆25Updated last year
- Guide Your Agent with Adaptive Multimodal Rewards (NeurIPS 2023 Accepted)☆32Updated last year
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆85Updated 11 months ago
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction (ICML 2023)☆39Updated last year
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆19Updated 5 months ago
- Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"☆54Updated 4 months ago
- ☆14Updated 2 years ago
- Code for Abstract-to-Executable Trajectory Translation for One Shot Task Generalization (ICML 2023)☆22Updated last year
- ☆27Updated 11 months ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆52Updated 7 months ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 3 months ago
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆66Updated 2 years ago
- (NeurIPS '22) LISA: Learning Interpretable Skill Abstractions - A framework for unsupervised skill learning using Imitation☆25Updated last year
- Code for the paper "Learning Options via Compression" at NeurIPS 2022☆22Updated last year
- A lightweight research framework☆21Updated 8 months ago
- ☆15Updated 2 years ago
- Reinforcement Learning via Supervised Learning☆68Updated 2 years ago
- PyTorch Package For Quasimetric Learning☆42Updated last week
- Code for the ICLR 2024 spotlight paper: "Learning to Act without Actions" (introducing Latent Action Policies)☆75Updated 3 months ago
- ☆28Updated last year
- The Controllable Agent project trains RL Agents able to optimize any reward function specified in real time, without any further learning…☆59Updated last year
- Implementation of ICML 2023 paper: Future-conditioned Unsupervised Pretraining for Decision Transformer☆27Updated last year
- ☆13Updated last year
- Official repository for paper "Versatile Offline Imitation from Observations and Examples via Regularized State-Occupancy Matching" (ICML…☆25Updated last year
- My Body Is A Cage☆38Updated 3 years ago
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆51Updated last month
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆33Updated 2 weeks ago