nathanwispinski / meta-rlLinks
A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.
☆17Updated 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
Sorting:
- PyTorch implementation of Episodic Meta Reinforcement Learning on variants of the "Two-Step" task. Reproduces the results found in three …☆35Updated 4 years ago
- ☆26Updated 4 years ago
- ☆40Updated 2 years ago
- Official implementation of the δ-model presented in the ICML 2024 paper "A Distributional Analogue to the Successor Representation".☆21Updated 6 months ago
- [NeurIPS 2021] Contrastive learning formulation of the active inference framework, for matching visual goal states.☆13Updated last year
- Code for the paper Deep Active Inference as Variational Policy Gradients.☆24Updated 5 years ago
- POPGym Library in JAX☆11Updated last year
- Tutorials on learning and using successor representations.☆52Updated 5 years ago
- Code and data for People construct simplified mental representations to plan☆23Updated last year
- 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 accompanying the paper "Would I have gotten that reward? Long-term credit assignment by counterfactual contribution analysis"☆11Updated last year
- Models of Sequential Decision-Making☆49Updated 4 months ago
- 🧊 🚩Comparison of active inference, q-learning and bayesian rl using modified FrozenLake environment☆34Updated 5 years ago
- Reinforcement Learning through Active Inference☆72Updated 5 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆17Updated 7 months ago
- Scalable Opponent Shaping Experiments in JAX☆24Updated last year
- Model-based reinforcement learning (generative simulator models and planning agents)☆16Updated 3 years ago
- PlaNet: Learning Latent Dynamics for Planning from Pixels☆10Updated 5 years ago
- Flexible modulation of sequence generation in the entorhinal-hippocampal system.☆13Updated 3 years ago
- ☆16Updated 2 years ago
- Free Energy Principle and Active Inference tutorials for newbies☆47Updated 4 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆21Updated 3 weeks ago
- NiceWebRL is a Python library for quickly making human subject experiments that leverage machine reinforcement learning environments.☆26Updated last week
- Accompanying code for "Learning and Planning in Average-Reward Markov Decision Processes"☆14Updated 4 years ago
- A Python Implementation of Predictive Coding (Whittington & Bogacz 2017)☆18Updated 4 years ago
- Implementation of an active inference capsule☆16Updated 4 years ago
- ☆18Updated last year
- Implementation of the two-step-task as described in "Prefrontal cortex as a meta-reinforcement learning system" and "Learning to Reinforc…☆58Updated 6 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- Agar.io for Continual Reinforcement Learning☆18Updated last week