World Models and Model-Based RL

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

WorldModelWrapper(*args, **kwargs)	World model wrapper.
DreamerActor(out_features[, depth, ...])	Dreamer actor network.
ObsEncoder([channels, num_layers, depth])	Observation encoder network.
ObsDecoder([channels, num_layers, ...])	Observation decoder network.
RSSMPosterior([hidden_dim, state_dim, scale_lb])	The posterior network of the RSSM.
RSSMPrior(action_spec[, hidden_dim, ...])	The prior network of the RSSM.
RSSMRollout(*args, **kwargs)	Rollout the RSSM network.