pyproject.toml
setup.cfg
setup.py
/Users/matbet/Downloads/supplementary_material_hetgps/code/torchrl/torchrl/csrc/pybind.cpp
build_tools/__init__.py
build_tools/setup_helpers/__init__.py
build_tools/setup_helpers/extension.py
torchrl/__init__.py
torchrl/_extension.py
torchrl/_utils.py
torchrl/version.py
torchrl.egg-info/PKG-INFO
torchrl.egg-info/SOURCES.txt
torchrl.egg-info/dependency_links.txt
torchrl.egg-info/not-zip-safe
torchrl.egg-info/requires.txt
torchrl.egg-info/top_level.txt
torchrl/collectors/__init__.py
torchrl/collectors/collectors.py
torchrl/collectors/utils.py
torchrl/collectors/distributed/__init__.py
torchrl/collectors/distributed/default_configs.py
torchrl/collectors/distributed/generic.py
torchrl/collectors/distributed/ray.py
torchrl/collectors/distributed/rpc.py
torchrl/collectors/distributed/sync.py
torchrl/collectors/distributed/utils.py
torchrl/data/__init__.py
torchrl/data/tensor_specs.py
torchrl/data/utils.py
torchrl/data/datasets/__init__.py
torchrl/data/datasets/atari_dqn.py
torchrl/data/datasets/common.py
torchrl/data/datasets/d4rl.py
torchrl/data/datasets/d4rl_infos.py
torchrl/data/datasets/gen_dgrl.py
torchrl/data/datasets/minari_data.py
torchrl/data/datasets/openml.py
torchrl/data/datasets/openx.py
torchrl/data/datasets/roboset.py
torchrl/data/datasets/utils.py
torchrl/data/datasets/vd4rl.py
torchrl/data/postprocs/__init__.py
torchrl/data/postprocs/postprocs.py
torchrl/data/replay_buffers/__init__.py
torchrl/data/replay_buffers/replay_buffers.py
torchrl/data/replay_buffers/samplers.py
torchrl/data/replay_buffers/storages.py
torchrl/data/replay_buffers/transforms.py
torchrl/data/replay_buffers/utils.py
torchrl/data/replay_buffers/writers.py
torchrl/data/rlhf/__init__.py
torchrl/data/rlhf/dataset.py
torchrl/data/rlhf/prompt.py
torchrl/data/rlhf/reward.py
torchrl/data/rlhf/utils.py
torchrl/envs/__init__.py
torchrl/envs/batched_envs.py
torchrl/envs/common.py
torchrl/envs/env_creator.py
torchrl/envs/gym_like.py
torchrl/envs/utils.py
torchrl/envs/vec_envs.py
torchrl/envs/libs/__init__.py
torchrl/envs/libs/_gym_utils.py
torchrl/envs/libs/brax.py
torchrl/envs/libs/dm_control.py
torchrl/envs/libs/envpool.py
torchrl/envs/libs/gym.py
torchrl/envs/libs/habitat.py
torchrl/envs/libs/isaacgym.py
torchrl/envs/libs/jax_utils.py
torchrl/envs/libs/jumanji.py
torchrl/envs/libs/openml.py
torchrl/envs/libs/pettingzoo.py
torchrl/envs/libs/robohive.py
torchrl/envs/libs/smacv2.py
torchrl/envs/libs/utils.py
torchrl/envs/libs/vmas.py
torchrl/envs/model_based/__init__.py
torchrl/envs/model_based/common.py
torchrl/envs/model_based/dreamer.py
torchrl/envs/transforms/__init__.py
torchrl/envs/transforms/functional.py
torchrl/envs/transforms/gym_transforms.py
torchrl/envs/transforms/r3m.py
torchrl/envs/transforms/rb_transforms.py
torchrl/envs/transforms/rlhf.py
torchrl/envs/transforms/transforms.py
torchrl/envs/transforms/utils.py
torchrl/envs/transforms/vc1.py
torchrl/envs/transforms/vip.py
torchrl/modules/__init__.py
torchrl/modules/distributions/__init__.py
torchrl/modules/distributions/continuous.py
torchrl/modules/distributions/discrete.py
torchrl/modules/distributions/truncated_normal.py
torchrl/modules/distributions/utils.py
torchrl/modules/models/__init__.py
torchrl/modules/models/decision_transformer.py
torchrl/modules/models/exploration.py
torchrl/modules/models/model_based.py
torchrl/modules/models/models.py
torchrl/modules/models/multiagent.py
torchrl/modules/models/rlhf.py
torchrl/modules/models/utils.py
torchrl/modules/planners/__init__.py
torchrl/modules/planners/cem.py
torchrl/modules/planners/common.py
torchrl/modules/planners/mppi.py
torchrl/modules/tensordict_module/__init__.py
torchrl/modules/tensordict_module/actors.py
torchrl/modules/tensordict_module/common.py
torchrl/modules/tensordict_module/exploration.py
torchrl/modules/tensordict_module/probabilistic.py
torchrl/modules/tensordict_module/rnn.py
torchrl/modules/tensordict_module/sequence.py
torchrl/modules/tensordict_module/world_models.py
torchrl/modules/utils/__init__.py
torchrl/modules/utils/mappings.py
torchrl/modules/utils/utils.py
torchrl/objectives/__init__.py
torchrl/objectives/a2c.py
torchrl/objectives/common.py
torchrl/objectives/cql.py
torchrl/objectives/ddpg.py
torchrl/objectives/decision_transformer.py
torchrl/objectives/deprecated.py
torchrl/objectives/dqn.py
torchrl/objectives/dreamer.py
torchrl/objectives/functional.py
torchrl/objectives/iql.py
torchrl/objectives/ppo.py
torchrl/objectives/redq.py
torchrl/objectives/reinforce.py
torchrl/objectives/sac.py
torchrl/objectives/td3.py
torchrl/objectives/utils.py
torchrl/objectives/multiagent/__init__.py
torchrl/objectives/multiagent/qmixer.py
torchrl/objectives/value/__init__.py
torchrl/objectives/value/advantages.py
torchrl/objectives/value/functional.py
torchrl/objectives/value/pg.py
torchrl/objectives/value/utils.py
torchrl/record/__init__.py
torchrl/record/recorder.py
torchrl/record/loggers/__init__.py
torchrl/record/loggers/common.py
torchrl/record/loggers/csv.py
torchrl/record/loggers/mlflow.py
torchrl/record/loggers/tensorboard.py
torchrl/record/loggers/utils.py
torchrl/record/loggers/wandb.py
torchrl/trainers/__init__.py
torchrl/trainers/trainers.py
torchrl/trainers/helpers/__init__.py
torchrl/trainers/helpers/collectors.py
torchrl/trainers/helpers/envs.py
torchrl/trainers/helpers/logger.py
torchrl/trainers/helpers/losses.py
torchrl/trainers/helpers/models.py
torchrl/trainers/helpers/replay_buffer.py
torchrl/trainers/helpers/trainers.py