NACLab / robust-active-inference
Official Implementation for the paper "R-AIF: Solving Sparse-Reward Robotic Tasks from Pixels with Active Inference and World Models"
☆32Updated 3 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for robust-active-inference
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆50Updated 5 months ago
- Implementation of an active inference capsule☆16Updated 3 years ago
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆52Updated last month
- Meta-RL Model-Based Algorithm☆27Updated 5 months ago
- ☆25Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆87Updated 11 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆63Updated 3 months ago
- ☆56Updated 2 years ago
- ☆55Updated last month
- Generalised UDRL☆37Updated 2 years ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆23Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆133Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆79Updated last year
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆83Updated 9 months ago
- ☆36Updated last year
- Highly scalable 2D JAX physics engine.☆35Updated last week
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆73Updated 6 months ago
- Goal-Conditioned Reinforcement Learning with JAX☆94Updated this week
- Foundation Policies with Hilbert Representations (ICML 2024)☆72Updated 7 months ago
- code for CoRL 2020 paper "Contrastive Variational Model-Based Reinforcement Learning for Complex Observations"☆24Updated 2 years ago
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆33Updated 8 months ago
- My Body Is A Cage☆38Updated 3 years ago
- ☆12Updated 8 months ago
- A lightweight research framework☆21Updated 8 months ago
- Code release for Efficient Planning in a Compact Latent Action Space (ICLR2023) https://arxiv.org/abs/2208.10291.☆98Updated last year
- Simplistic Pytorch Implementation of the Dreamer-RL☆20Updated 2 years ago
- A PyTorch implementation of Implicit Behavioral Cloning☆94Updated 2 years ago
- ☆28Updated 2 years ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆22Updated last year
- Extreme Q-Learning: Max Entropy RL without Entropy☆80Updated last year