There are 18 repositories under model-based-reinforcement-learning topic.
OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.
A curated list of awesome model based RL resources (continually updated)
Library for Model Based RL
Personal notes about scientific and research works on "Decision-Making for Autonomous Driving"
Code to reproduce the experiments in Sample Efficient Reinforcement Learning via Model-Ensemble Exploration and Exploitation (MEEE).
DI-engine docs (Chinese and English)
Unofficial Pytorch code for "Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Models"
Official repository for "iVideoGPT: Interactive VideoGPTs are Scalable World Models" (NeurIPS 2024), https://arxiv.org/abs/2405.15223
Unofficial Implementation of the paper "Data-Efficient Reinforcement Learning with Probabilistic Model Predictive Control", applied to gym environments
Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch
(Experimental, a lot of bugs) Automatic fingering generator for piano scores, determining optimal fingering using Model-Based Reinforcement Learning, written in the Julia language.
Model-based Reinforcement Learning Framework
Code release for Efficient Planning in a Compact Latent Action Space (ICLR2023) https://arxiv.org/abs/2208.10291.
A curated list of awesome Model-based reinforcement learning resources
Deep active inference agents using Monte-Carlo methods
Implementation of Danijar's latest iteration for his Dreamer line of work
Latent Imagination Facilitates Zero-Shot Transfer in Autonomous Racing
Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024
Code for "World Model as a Graph: Learning Latent Landmarks for Planning" (ICML 2021 Long Presentation)
An implementation of MuZero in JAX.
Adaptable tools to make reinforcement learning and evolutionary computation algorithms.
Model-based reinforcement learning in TensorFlow
Code release for "HarmonyDream: Task Harmonization Inside World Models" (ICML 2024), https://arxiv.org/abs/2310.00344
这是一个关于基于模型的强化学习的资料,包括一些代码地址、paper、slide等。
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environments.
The Hierarchical Intrinsically Motivated Agent (HIMA) is an algorithm that is intended to exhibit an adaptive goal-directed behavior using neurophysiological models of the neocortex, basal ganglia, and thalamus.
LAMBDA is a model-based reinforcement learning agent that uses Bayesian world models for safe policy optimization
Official codebase for "Privileged Sensing Scaffolds Reinforcement Learning", contains the Scaffolder algorithm and Sensory Scaffolding Suite.
This repository has code for the paper "Model-based Safe Deep Reinforcement Learning via a Constrained Proximal Policy Optimization Algorithm" accepted at NeurIPS 2022.
[ICLR 2025] Official implementation of DICL (Disentangled In-Context Learning), featured in the paper "Zero-shot Model-based Reinforcement Learning using Large Language Models".
Official PyTorch implementation of "DC-MPC: Discrete Codebook Model Predictive Control"
Control simulation of a mass-spring-damper system using a model-based reinforcement learning algorithm