#Model-based-reinforcement-learning

opendilab

DI-engine

OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.

pytorch

rl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

lucidrains

dreamer4

Implementation of Danijar's latest iteration for his Dreamer line of work

lucidrains

improving-transformers-world-model-for-rl

Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch

CPS-TUWien

racing_dreamer

Latent Imagination Facilitates Zero-Shot Transfer in Autonomous Racing

AlejandroMllo

action_flow_matching

Robotics: Science and Systems (RSS) 2025 | Action Flow Matching for Continual Robot Learning | Online and Non-episodic Robot Dynamics Model Refinement.