NACLab / robust-active-inference
Official Implementation for the paper "R-AIF: Solving Sparse-Reward Robotic Tasks from Pixels with Active Inference and World Models"
☆31Updated 3 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for robust-active-inference
- Foundation Policies with Hilbert Representations (ICML 2024)☆72Updated 6 months ago
- Implementation of an active inference capsule☆16Updated 3 years ago
- PyTorch Package For Quasimetric Learning☆42Updated last week
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆11Updated 4 years ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆50Updated 5 months ago
- ☆25Updated 11 months ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆23Updated last year
- Contains JAX implementation of algorithms for inverse reinforcement learning☆63Updated 2 months ago
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆51Updated last month
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆57Updated last year
- ☆53Updated 3 weeks ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆87Updated 11 months ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 3 months ago
- ☆24Updated 3 years ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆22Updated last year
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆30Updated last year
- ☆36Updated last year
- A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.☆14Updated last year
- ☆12Updated 7 months ago
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆12Updated 4 months ago
- ☆22Updated 4 months ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆52Updated 7 months ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆18Updated 2 years ago
- ☆62Updated 2 months ago
- Generalised UDRL☆37Updated 2 years ago
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆27Updated last year
- Intrinsic Reward Matching (IRM) implementation (from Adeniji and Xie et al 2022)☆43Updated 10 months ago
- ☆20Updated last year
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆25Updated last year
- flexible meta-learning in jax☆12Updated last year