Search Packages

torchrl

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

torchrl-nightly

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

dreamer4

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

di-engine

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

improving-transformers-world-model

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

mbrl

A PyTorch library for model-based reinforcement learning research

deepcubeai

Learning Discrete World Models for Heuristic Search

bellman

Model-based reinforcement learning in TensorFlow

baconian

model-based reinforcement learning toolbox