model-based-reinforcement-learning topic
jax_muzero
An implementation of MuZero in JAX.
latentplan
Code release for Efficient Planning in a Compact Latent Action Space (ICLR2023) https://arxiv.org/abs/2208.10291.
omd
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
omega
A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environments.
VSG
Learning Robust Dynamics Through Variational Sparse Gating
rl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
rosmo
Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023
MEEE
Code to reproduce the experiments in Sample Efficient Reinforcement Learning via Model-Ensemble Exploration and Exploitation (MEEE).
PianoFingering.jl
(Experimental, a lot of bugs) Advanced automatic fingering generator for piano scores, determining optimal fingering using Model-Based Reinforcement Learning, written in the Julia language.
Pearl
Adaptable tools to make reinforcement learning and evolutionary computation algorithms.