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
- ☆25Updated 4 years 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
- Code for the paper Deep Active Inference as Variational Policy Gradients.☆24Updated 5 years ago
- POPGym Library in JAX☆11Updated 11 months ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆17Updated 4 months ago
- PyTorch implementation of Episodic Meta Reinforcement Learning on variants of the "Two-Step" task. Reproduces the results found in three …☆35Updated 4 years ago
- Official implementation of the δ-model presented in the ICML 2024 paper "A Distributional Analogue to the Successor Representation".☆20Updated 4 months ago
- Code accompanying the paper "Would I have gotten that reward? Long-term credit assignment by counterfactual contribution analysis"☆11Updated last year
- ☆40Updated 2 years ago
- PlaNet: Learning Latent Dynamics for Planning from Pixels☆10Updated 5 years ago
- Tutorials on learning and using successor representations.☆50Updated 5 years ago
- Deep active inference agents using Monte-Carlo methods☆87Updated 3 years ago
- [NeurIPS 2021] Contrastive learning formulation of the active inference framework, for matching visual goal states.☆13Updated last year
- ☆16Updated 2 years ago
- Model-based reinforcement learning (generative simulator models and planning agents)☆15Updated 3 years ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- Generalised UDRL☆37Updated 2 years ago
- 🧊 🚩Comparison of active inference, q-learning and bayesian rl using modified FrozenLake environment☆34Updated 5 years ago
- A PyTorch Implementation of Skipper☆20Updated 5 months ago
- Code that can be used to reproduce the experiments in our paper "Estimating Risk and Uncertainty in Deep Reinforcement Learning"☆30Updated 2 years ago
- Code and data for People construct simplified mental representations to plan☆23Updated last year
- Implementation of Proximal Policy Optimization in Jax+Flax☆18Updated last year
- Episodic Control☆19Updated 2 years ago
- Scalable Opponent Shaping Experiments in JAX☆24Updated 11 months ago
- Code for Model-Free Opponent Shaping (ICML 2022)☆18Updated 2 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Updated 2 years ago
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆26Updated last year
- A Python Implementation of Predictive Coding (Whittington & Bogacz 2017)☆15Updated 4 years ago
- ☆20Updated 2 years ago