JesseFarebro / distributional-sr
Official implementation of the δ-model presented in the ICML 2024 paper "A Distributional Analogue to the Successor Representation".
☆17Updated 3 months ago
Alternatives and similar repositories for distributional-sr:
Users that are interested in distributional-sr are comparing it to the libraries listed below
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"☆43Updated 7 months ago
- Code for Model-Free Opponent Shaping (ICML 2022)☆16Updated 2 years ago
- ☆25Updated 4 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆70Updated 5 months ago
- ☆70Updated 5 months ago
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆85Updated 3 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆140Updated last year
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆17Updated 3 months ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆14Updated 2 years ago
- A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.☆15Updated last year
- POPGym Library in JAX☆11Updated 10 months ago
- PAIRED in PyTorch 🔥☆58Updated last year
- Baselines for gymnax 🤖☆63Updated last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆53Updated 2 years ago
- ☆19Updated 10 months ago
- ☆42Updated last year
- Reinforcement Learning through Active Inference☆69Updated 4 years ago
- Learning diverse options through the Laplacian representation.☆23Updated last year
- Highly scalable 2D JAX physics engine.☆44Updated 2 weeks ago
- Benchmarking RL generalization in an interpretable way.☆142Updated last week
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated last year
- Accompanying code for "Learning and Planning in Average-Reward Markov Decision Processes"☆14Updated 4 years ago
- Tutorials on learning and using successor representations.☆50Updated 5 years ago
- ☆11Updated 11 months ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆94Updated last year
- Scalable Opponent Shaping Experiments in JAX☆24Updated 10 months ago
- Simple JAX Graphics Library.☆29Updated 3 months ago
- Code and data for People construct simplified mental representations to plan☆23Updated last year
- An Open-Ended Agentic Simulator☆39Updated 6 months ago
- ☆41Updated last year