CognitiveModeling / THICK
☆16Updated 10 months ago
Alternatives and similar repositories for THICK:
Users that are interested in THICK are comparing it to the libraries listed below
- ☆22Updated 2 years ago
- Continual RL with wold models☆13Updated last year
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆34Updated 11 months ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆23Updated last year
- Skeleton for scalable and flexible Jax RL implementations☆69Updated last year
- Learning Robust Dynamics Through Variational Sparse Gating☆21Updated 2 years ago
- EARL: Environment for Autonomous Reinforcement Learning☆34Updated 2 years ago
- Official release of CompoSuite, a compositional RL benchmark☆47Updated last year
- ☆46Updated 2 years ago
- This repository is a collection of widely used self-supervised auxiliary losses used for learning representations in reinforcement learni…☆14Updated last year
- ☆23Updated 2 years ago
- Discovering and Achieving Goals via World Models, NeurIPS 2021☆85Updated last year
- Official Codebase for Offline Reinforcement Learning from Images with Latent Space Models☆31Updated 3 years ago
- Efficient seed-parallel implementation of "Breaking the Replay Ratio Barrier"☆22Updated last year
- ☆14Updated 9 months ago
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆32Updated last year
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆60Updated 7 months ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆31Updated last year
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆29Updated 5 months ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20Updated 2 years ago
- ☆24Updated 7 months ago
- ☆41Updated last year
- Object Centric Atari games☆63Updated this week
- Code for "On Uncertainty in Deep State Space Models for Model-Based Reinforcement Learning" (TMLR, 2022)☆14Updated last year
- A version of the DeepMind Control Suite with randomly generated graphics, for measuring visual generalization in continuous control.☆17Updated 4 years ago
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆29Updated 2 years ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Updated 9 months ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆23Updated last year
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆54Updated 10 months ago