.. currentmodule:: torchrl.modules

World Models and Model-Based RL
===============================

Modules for model-based reinforcement learning, including world models and dynamics models.

.. autosummary::
    :toctree: generated/
    :template: rl_template_noinherit.rst

    WorldModelWrapper
    DreamerActor
    ObsEncoder
    ObsDecoder
    RSSMPosterior
    RSSMPrior
    RSSMRollout