seohongpark / PMA
Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)
☆32Updated 2 years ago
Alternatives and similar repositories for PMA:
Users that are interested in PMA are comparing it to the libraries listed below
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆53Updated 11 months ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆64Updated last year
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆24Updated last year
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆77Updated 3 months ago
- Lipschitz-constrained Unsupervised Skill Discovery (ICLR 2022)☆36Updated last year
- Pre-Training Goal-based Models for Sample-Efficient Reinforcement Learning☆25Updated last year
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 7 months ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆77Updated 9 months ago
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆37Updated last year
- ☆15Updated last year
- [ICLR 2023] Choreographer: a model-based agent that discovers and learns unsupervised skills in latent imagination, and it's able to effi…☆40Updated 8 months ago
- EARL: Environment for Autonomous Reinforcement Learning☆36Updated 2 years ago
- Implementation of Jump-Start Reinforcement Learning (JSRL) with Stable Baselines3☆31Updated last year
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Updated 10 months ago
- ☆14Updated 2 years ago
- Authors' PyTorch implementation of 'Recomposing the Reinforcement Learning Building-Blocks with Hypernetworks' (HypeRL)☆25Updated 3 years ago
- Official Codebase for Offline Reinforcement Learning from Images with Latent Space Models☆30Updated 3 years ago
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆55Updated 5 months ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆32Updated last year
- Reinforcement Learning via Supervised Learning☆71Updated 2 years ago
- Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022☆43Updated last year
- ☆41Updated 3 months ago
- My Body Is A Cage☆39Updated 3 years ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆85Updated last year
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆31Updated last year
- Intrinsic Reward Matching (IRM) implementation (from Adeniji and Xie et al 2022)☆43Updated last year
- Foundation Policies with Hilbert Representations (ICML 2024)☆79Updated 10 months ago
- ☆22Updated 3 years ago
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆99Updated 9 months ago
- ☆15Updated 3 years ago