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
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆85Updated last year
- ☆15Updated last year
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆29Updated 3 months ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆52Updated 7 months ago
- PyTorch code accompanying the paper "Imitating Graph-Based Planning with Goal-Conditioned Policies" (ICLR 2023).☆19Updated last year
- ☆28Updated last year
- Clean, extensible implementation of MACAW [ICML 2021]☆11Updated 2 years ago
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆52Updated last month
- Guide Your Agent with Adaptive Multimodal Rewards (NeurIPS 2023 Accepted)☆32Updated last year
- Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022☆44Updated 8 months ago
- ☆25Updated last year
- ☆15Updated 2 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- Official codebase for Exact Energy-Guided Diffusion Sampling via Contrastive Energy Prediction (ICML 2023)☆39Updated last year
- Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"☆54Updated 4 months ago
- [NeurIPS 2023] Efficient Diffusion Policy☆82Updated last year
- ☆13Updated 7 months ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆76Updated last year
- Open source code for paper "Optimal Goal-Reaching Reinforcement Learning via Quasimetric Learning" ICML 2023☆44Updated 6 months ago
- Code for "World Model as a Graph: Learning Latent Landmarks for Planning" (ICML 2021 Long Presentation)☆64Updated 3 years ago
- Code for the paper "Inference via Interpolation: Contrastive Representations Provably Enable Planning and Inference"☆37Updated 4 months ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆41Updated last year
- ☆11Updated 2 years ago
- A System for Morphology-Task Generalization via Unified Representation and Behavior Distillation (ICLR2023)☆12Updated last year
- My Body Is A Cage☆38Updated 3 years ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆19Updated 5 months ago
- Official Codebase for Offline Reinforcement Learning from Images with Latent Space Models☆29Updated 3 years ago
- [NeurIPS 2024] GenRL: Multimodal foundation world models allow grounding language and video prompts into embodied domains, by turning the…☆59Updated 3 months ago
- Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024☆59Updated 8 months ago