Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper
☆21Jul 14, 2024Updated last year
Alternatives and similar repositories for Hieros
Users that are interested in Hieros are comparing it to the libraries listed below
Sorting:
- Implementation of Neurips 2023 Paper "Multi Time Scale World Models"☆16Nov 8, 2024Updated last year
- ☆29Mar 20, 2024Updated 2 years ago
- POPGym Library in JAX☆12Apr 15, 2024Updated last year
- Giving Up Control: Neurons as Reinforcement Learning Agents☆13May 6, 2024Updated last year
- Official repository for our paper on "Action Inference by Maximising Evidence: Zero-Shot Imitation from Observation with World Models"☆13Dec 4, 2023Updated 2 years ago
- A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.☆18Feb 27, 2023Updated 3 years ago
- IPHYRE: Interactive Physical Reasoning, ICLR 2024☆19Oct 18, 2024Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆165Jun 23, 2023Updated 2 years ago
- Transformer-based World Models☆89Apr 4, 2023Updated 2 years ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆18Jan 16, 2023Updated 3 years ago
- Pytorch Implementation of Learning Latent Dynamic Robust Representations for World Models☆25May 11, 2024Updated last year
- PlaNet: Learning Latent Dynamics for Planning from Pixels☆10Feb 13, 2020Updated 6 years ago
- Animal-AI supports interdisciplinary research to help better understand human, animal, and artificial cognition.☆87Oct 4, 2025Updated 5 months ago
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Jan 7, 2025Updated last year
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆79Jun 4, 2024Updated last year
- Open-source codebase for MAZero, from "Efficient Multi-agent Reinforcement Learning by Planning" at ICLR 2024.☆42May 8, 2024Updated last year
- ☆11Jun 7, 2023Updated 2 years ago
- Adding Dreamer-v3's implementation tricks to CleanRL's PPO☆14May 19, 2023Updated 2 years ago
- Code for the paper: Sparsely Changing Latent States for Prediction and Planning in Partially Observable Domains☆10Nov 12, 2021Updated 4 years ago
- ☆13Mar 13, 2024Updated 2 years ago
- ☆69Oct 22, 2024Updated last year
- SGD (stochastic gradient descent) with OGR - online gradient regression Hessian estimator☆16May 11, 2023Updated 2 years ago
- Codebase for project about unsupervised skill learning via variational inference and causality.☆43Oct 1, 2023Updated 2 years ago
- Code for the paper "Harnessing Discrete Representations for Continual Reinforcement Learning"☆15Jun 16, 2024Updated last year
- Minimizing Control for Credit Assignment with Strong Feedback☆14Nov 3, 2024Updated last year
- Extending rllab to event-driven multiagent environments☆13Oct 1, 2018Updated 7 years ago
- ☆12Jul 8, 2024Updated last year
- ☆130Feb 22, 2025Updated last year
- ☆13Sep 24, 2022Updated 3 years ago
- Contrastive UCB: Provably Efficient Contrastive Self-Supervised Learning in Online Reinforcement Learning☆11Jun 16, 2022Updated 3 years ago
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆29Jun 2, 2025Updated 9 months ago
- An implementation of a neural network training routine using derivative information in Pytorch.☆10Dec 19, 2020Updated 5 years ago
- The official repository for our paper "The Neural Data Router: Adaptive Control Flow in Transformers Improves Systematic Generalization".☆34Jun 11, 2025Updated 9 months ago
- A Simplified Pytorch Version of the Dreamer Algorithm☆152Jul 24, 2023Updated 2 years ago
- ☆11Nov 27, 2025Updated 3 months ago
- ☆19Apr 12, 2024Updated last year
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆12Aug 31, 2020Updated 5 years ago
- ☆316Jan 8, 2025Updated last year
- Clockwork Variational Autoencoder☆50Jun 30, 2021Updated 4 years ago