nathanwispinski / meta-rl
A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.
☆15Updated 2 years ago
Alternatives and similar repositories for meta-rl:
Users that are interested in meta-rl are comparing it to the libraries listed below
- Official implementation of the δ-model presented in the ICML 2024 paper "A Distributional Analogue to the Successor Representation".☆20Updated 4 months ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆15Updated 2 years ago
- POPGym Library in JAX☆11Updated 10 months ago
- ☆25Updated 4 years ago
- ☆16Updated 2 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆17Updated 4 months ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- Tutorials on learning and using successor representations.☆50Updated 5 years ago
- PlaNet: Learning Latent Dynamics for Planning from Pixels☆10Updated 5 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆21Updated 2 years ago
- ☆35Updated 2 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Updated 2 years ago
- Model-based reinforcement learning (generative simulator models and planning agents)☆15Updated 3 years ago
- PyTorch implementation of Episodic Meta Reinforcement Learning on variants of the "Two-Step" task. Reproduces the results found in three …☆34Updated 4 years ago
- 🧊 🚩Comparison of active inference, q-learning and bayesian rl using modified FrozenLake environment☆34Updated 5 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆69Updated 6 months ago
- Minimizing Control for Credit Assignment with Strong Feedback☆14Updated 4 months ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆24Updated last year
- GPT implementation in Flax☆18Updated 3 years ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆14Updated 2 years ago
- Accompanying code for "Learning and Planning in Average-Reward Markov Decision Processes"☆14Updated 4 years ago
- PyTorch Package For Quasimetric Learning☆41Updated 4 months ago
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆27Updated 8 months ago
- ☆20Updated 2 years ago
- PushWorld: A benchmark for manipulation planning with tools and movable obstacles☆82Updated 10 months ago
- Code and data for People construct simplified mental representations to plan☆23Updated last year
- Episodic Control☆19Updated 2 years ago
- Code for the paper Deep Active Inference as Variational Policy Gradients.☆24Updated 5 years ago
- ☆16Updated 11 months ago