All modules for which code is available
- pyvers.implement_for
- tensordict.nn.distributions.continuous
- torchrl._utils
- torchrl.collectors._base
- torchrl.collectors._multi_async
- torchrl.collectors._multi_base
- torchrl.collectors._multi_sync
- torchrl.collectors._single
- torchrl.collectors._single_async
- torchrl.collectors.distributed.generic
- torchrl.collectors.distributed.ray
- torchrl.collectors.distributed.rpc
- torchrl.collectors.distributed.sync
- torchrl.collectors.distributed.utils
- torchrl.collectors.llm.base
- torchrl.collectors.llm.ray_collector
- torchrl.collectors.llm.weight_update.vllm
- torchrl.collectors.llm.weight_update.vllm_v2
- torchrl.collectors.utils
- torchrl.collectors.weight_update
- torchrl.data.datasets.atari_dqn
- 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.llm.history
- torchrl.data.llm.topk
- torchrl.data.replay_buffers.checkpointers
- torchrl.data.replay_buffers.ray_buffer
- torchrl.data.replay_buffers.replay_buffers
- torchrl.data.replay_buffers.samplers
- torchrl.data.replay_buffers.storages
- torchrl.data.replay_buffers.writers
- torchrl.data.tensor_specs
- torchrl.envs.async_envs
- torchrl.envs.batched_envs
- torchrl.envs.common
- torchrl.envs.custom.chess
- torchrl.envs.custom.llm
- torchrl.envs.custom.pendulum
- torchrl.envs.custom.tictactoeenv
- 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.isaac_lab
- torchrl.envs.libs.isaacgym
- torchrl.envs.libs.jumanji
- torchrl.envs.libs.meltingpot
- torchrl.envs.libs.openml
- torchrl.envs.libs.openspiel
- torchrl.envs.libs.pettingzoo
- torchrl.envs.libs.procgen
- torchrl.envs.libs.robohive
- torchrl.envs.libs.smacv2
- torchrl.envs.libs.unity_mlagents
- torchrl.envs.libs.vmas
- torchrl.envs.llm.chat
- torchrl.envs.llm.datasets.gsm8k
- torchrl.envs.llm.datasets.ifeval
- torchrl.envs.llm.envs
- torchrl.envs.llm.libs.mlgym
- torchrl.envs.llm.reward.gsm8k
- torchrl.envs.llm.reward.ifeval._scorer
- torchrl.envs.llm.transforms.browser
- torchrl.envs.llm.transforms.dataloading
- torchrl.envs.llm.transforms.format
- torchrl.envs.llm.transforms.kl
- torchrl.envs.llm.transforms.policy_version
- torchrl.envs.llm.transforms.reason
- torchrl.envs.llm.transforms.tokenizer
- torchrl.envs.llm.transforms.tools
- torchrl.envs.model_based.common
- torchrl.envs.model_based.dreamer
- torchrl.envs.transforms.gym_transforms
- torchrl.envs.transforms.module
- torchrl.envs.transforms.r3m
- torchrl.envs.transforms.transforms
- torchrl.envs.transforms.vc1
- torchrl.envs.transforms.vecnorm
- torchrl.envs.transforms.vip
- torchrl.envs.utils
- torchrl.modules.distributions.continuous
- torchrl.modules.distributions.discrete
- torchrl.modules.llm.backends.vllm.vllm_async
- torchrl.modules.llm.backends.vllm.vllm_sync
- torchrl.modules.llm.backends.vllm.vllm_utils
- torchrl.modules.llm.policies.common
- torchrl.modules.llm.policies.transformers_wrapper
- torchrl.modules.llm.policies.vllm_wrapper
- torchrl.modules.mcts.scores
- torchrl.modules.models.decision_transformer
- torchrl.modules.models.exploration
- torchrl.modules.models.model_based
- torchrl.modules.models.models
- torchrl.modules.models.utils
- 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.objectives.a2c
- torchrl.objectives.common
- torchrl.objectives.cql
- torchrl.objectives.crossq
- torchrl.objectives.ddpg
- torchrl.objectives.decision_transformer
- torchrl.objectives.dqn
- torchrl.objectives.dreamer
- torchrl.objectives.gail
- torchrl.objectives.iql
- torchrl.objectives.llm.grpo
- torchrl.objectives.llm.sft
- torchrl.objectives.ppo
- torchrl.objectives.redq
- torchrl.objectives.reinforce
- torchrl.objectives.sac
- torchrl.objectives.td3
- torchrl.objectives.td3_bc
- torchrl.objectives.utils
- torchrl.objectives.value.advantages
- torchrl.record.loggers.common
- torchrl.record.loggers.csv
- torchrl.record.loggers.mlflow
- torchrl.record.loggers.tensorboard
- torchrl.record.loggers.trackio
- torchrl.record.loggers.utils
- torchrl.record.loggers.wandb
- torchrl.record.recorder
- torchrl.services
- torchrl.trainers.algorithms.configs.collectors
- torchrl.trainers.algorithms.configs.common
- torchrl.trainers.algorithms.configs.data
- torchrl.trainers.algorithms.configs.envs
- torchrl.trainers.algorithms.configs.envs_libs
- torchrl.trainers.algorithms.configs.logging
- torchrl.trainers.algorithms.configs.modules
- torchrl.trainers.algorithms.configs.objectives
- torchrl.trainers.algorithms.configs.trainers
- torchrl.trainers.algorithms.configs.transforms
- torchrl.trainers.algorithms.configs.utils
- torchrl.trainers.algorithms.ppo
- torchrl.trainers.algorithms.sac
- torchrl.trainers.helpers.collectors
- torchrl.trainers.helpers.envs
- torchrl.trainers.helpers.losses
- torchrl.trainers.helpers.replay_buffer
- torchrl.trainers.helpers.trainers
- torchrl.trainers.trainers
- torchrl.weight_update._distributed
- torchrl.weight_update._mp
- torchrl.weight_update._noupdate
- torchrl.weight_update._ray
- torchrl.weight_update._rpc
- torchrl.weight_update._shared
- torchrl.weight_update.llm.vllm_double_buffer
- torchrl.weight_update.llm.vllm_nccl
- torchrl.weight_update.weight_sync_schemes