JesseFarebro / distributional-sr
Official implementation of the δ-model presented in the ICML 2024 paper "A Distributional Analogue to the Successor Representation".
☆14Updated this week
Related projects ⓘ
Alternatives and complementary repositories for distributional-sr
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"☆42Updated 4 months ago
- Code for Model-Free Opponent Shaping (ICML 2022)☆16Updated last year
- ☆61Updated 2 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆62Updated 2 months ago
- Tutorials on learning and using successor representations.☆50Updated 5 years ago
- A collection of RL algorithms written in JAX.☆94Updated 2 years ago
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆83Updated 3 years ago
- PAIRED in PyTorch 🔥☆56Updated last year
- An Open-Ended Agentic Simulator☆22Updated 2 months ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆12Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆49Updated 2 years ago
- Baselines for gymnax 🤖☆58Updated last year
- ☆36Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆132Updated last year
- ☆28Updated 3 years ago
- ☆19Updated 6 months ago
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- Conservative Q learning in Jax☆50Updated last year
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆52Updated 9 months ago
- Scalable Opponent Shaping Experiments in JAX☆21Updated 6 months ago
- 🧊 🚩Comparison of active inference, q-learning and bayesian rl using modified FrozenLake environment☆35Updated 4 years ago
- Sandbox environment for generalizable agent research☆22Updated 2 years ago
- TeachMyAgent is a testbed platform for Automatic Curriculum Learning methods in Deep RL.☆66Updated 11 months ago
- A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.☆14Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆86Updated 11 months ago
- Code and data for People construct simplified mental representations to plan☆21Updated last year
- ExORL: Exploratory Data for Offline Reinforcement Learning☆104Updated 2 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆14Updated last week
- JAX implementation of RL algorithms and vectorized environments☆34Updated 10 months ago
- Benchmarking RL generalization in an interpretable way.☆131Updated 8 months ago