OpenDILab Decision AI Engine
-
Updated
Jul 3, 2024 - Python
OpenDILab Decision AI Engine
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Library for Model Based RL
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"
Latent Imagination Facilitates Zero-Shot Transfer in Autonomous Racing
Unofficial Implementation of the paper "Data-Efficient Reinforcement Learning with Probabilistic Model Predictive Control", applied to gym environments
Deep active inference agents using Monte-Carlo methods
LAMBDA is a model-based reinforcement learning agent that uses Bayesian world models for safe policy optimization
Code release for Efficient Planning in a Compact Latent Action Space (ICLR2023) https://arxiv.org/abs/2208.10291.
Model-based Reinforcement Learning Framework
An implementation of MuZero in JAX.
Model-based reinforcement learning in TensorFlow
A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environments.
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)
Learning Robust Dynamics Through Variational Sparse Gating
Code for Asynchronous Methods for Model-Based Reinforcement Learning
This repository has code for the paper "Model-based Safe Deep Reinforcement Learning via a Constrained Proximal Policy Optimization Algorithm" accepted at NeurIPS 2022.
Add a description, image, and links to the model-based-reinforcement-learning topic page so that developers can more easily learn about it.
To associate your repository with the model-based-reinforcement-learning topic, visit your repo's landing page and select "manage topics."