All modules for which code is available

• tensordict.nn.utils
• torchrl._utils
• torchrl.collectors.collectors
• torchrl.collectors.distributed.generic
• torchrl.collectors.distributed.ray
• torchrl.collectors.distributed.rpc
• torchrl.collectors.distributed.sync
• torchrl.collectors.distributed.utils
• torchrl.collectors.utils
• torchrl.data.datasets.atari_dqn
• torchrl.data.datasets.common
• torchrl.data.datasets.d4rl
• torchrl.data.datasets.gen_dgrl
• torchrl.data.datasets.minari_data
• torchrl.data.datasets.openml
• torchrl.data.datasets.openx
• torchrl.data.datasets.roboset
• torchrl.data.datasets.vd4rl
• torchrl.data.postprocs.postprocs
• torchrl.data.replay_buffers.replay_buffers
• torchrl.data.replay_buffers.samplers
• torchrl.data.replay_buffers.storages
• torchrl.data.replay_buffers.writers
• torchrl.data.rlhf.dataset
• torchrl.data.rlhf.prompt
• torchrl.data.rlhf.reward
• torchrl.data.rlhf.utils
• torchrl.data.tensor_specs
• torchrl.data.utils
• torchrl.envs.batched_envs
• torchrl.envs.common
• torchrl.envs.env_creator
• torchrl.envs.gym_like
• torchrl.envs.libs.brax
• torchrl.envs.libs.dm_control
• torchrl.envs.libs.envpool
• torchrl.envs.libs.gym
• torchrl.envs.libs.habitat
• torchrl.envs.libs.isaacgym
• torchrl.envs.libs.jumanji
• torchrl.envs.libs.meltingpot
• torchrl.envs.libs.openml
• torchrl.envs.libs.pettingzoo
• torchrl.envs.libs.robohive
• torchrl.envs.libs.smacv2
• torchrl.envs.libs.vmas
• torchrl.envs.model_based.common
• torchrl.envs.model_based.dreamer
• torchrl.envs.transforms.gym_transforms
• torchrl.envs.transforms.r3m
• torchrl.envs.transforms.rb_transforms
• torchrl.envs.transforms.rlhf
• torchrl.envs.transforms.transforms
• torchrl.envs.transforms.vc1
• torchrl.envs.transforms.vip
• torchrl.envs.utils
• torchrl.modules.distributions.continuous
• torchrl.modules.distributions.discrete
• torchrl.modules.models.decision_transformer
• torchrl.modules.models.exploration
• torchrl.modules.models.model_based
• torchrl.modules.models.models
• torchrl.modules.models.multiagent
• torchrl.modules.models.utils
• torchrl.modules.planners.cem
• torchrl.modules.planners.common
• torchrl.modules.planners.mppi
• torchrl.modules.tensordict_module.actors
• torchrl.modules.tensordict_module.common
• torchrl.modules.tensordict_module.exploration
• torchrl.modules.tensordict_module.probabilistic
• torchrl.modules.tensordict_module.rnn
• torchrl.modules.tensordict_module.sequence
• torchrl.modules.tensordict_module.world_models
• torchrl.modules.utils.mappings
• torchrl.objectives.a2c
• torchrl.objectives.common
• torchrl.objectives.cql
• torchrl.objectives.ddpg
• torchrl.objectives.decision_transformer
• torchrl.objectives.dqn
• torchrl.objectives.dreamer
• torchrl.objectives.iql
• torchrl.objectives.multiagent.qmixer
• torchrl.objectives.ppo
• torchrl.objectives.redq
• torchrl.objectives.reinforce
• torchrl.objectives.sac
• torchrl.objectives.td3
• torchrl.objectives.utils
• torchrl.objectives.value.advantages
• torchrl.objectives.value.functional
• torchrl.record.loggers.common
• torchrl.record.loggers.csv
• torchrl.record.loggers.mlflow
• torchrl.record.loggers.tensorboard
• torchrl.record.loggers.utils
• torchrl.record.loggers.wandb
• torchrl.record.recorder
• torchrl.trainers.helpers.collectors
• torchrl.trainers.helpers.envs
• torchrl.trainers.helpers.losses
• torchrl.trainers.helpers.replay_buffer
• torchrl.trainers.helpers.trainers
• torchrl.trainers.trainers