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
- Tutorials on learning and using successor representations.☆52Updated 5 years ago
- POPGym Library in JAX☆11Updated last year
- Official implementation of the δ-model presented in the ICML 2024 paper "A Distributional Analogue to the Successor Representation".☆20Updated 5 months ago
- ☆26Updated 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
- ☆40Updated 2 years ago
- Code accompanying the paper "Would I have gotten that reward? Long-term credit assignment by counterfactual contribution analysis"☆11Updated last year
- PlaNet: Learning Latent Dynamics for Planning from Pixels☆10Updated 5 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆21Updated 2 years ago
- Episodic Control☆19Updated 2 years ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆17Updated 5 months ago
- Code and data for People construct simplified mental representations to plan☆23Updated last year
- Python library for easily making web Apps to compare humans and AI☆19Updated this week
- ☆16Updated 2 years 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
- Generalised UDRL☆37Updated 2 years ago
- ☆20Updated 2 years ago
- PyTorch Package For Quasimetric Learning☆41Updated 5 months ago
- 🧊 🚩Comparison of active inference, q-learning and bayesian rl using modified FrozenLake environment☆34Updated 5 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- Scalable Opponent Shaping Experiments in JAX☆24Updated last year
- Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"☆44Updated last year
- Code for the paper Deep Active Inference as Variational Policy Gradients.☆24Updated 5 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆9Updated last year
- Accompanying code for "Learning and Planning in Average-Reward Markov Decision Processes"☆14Updated 4 years ago
- Model-based reinforcement learning (generative simulator models and planning agents)☆15Updated 3 years ago
- Implementation of Proximal Policy Optimization in Jax+Flax☆18Updated last year
- Deep active inference agents using Monte-Carlo methods☆88Updated 3 years ago
- ☆35Updated 2 years ago