Path	Lines of Code
config/__init__.py	1
config/agent/components/gradnorm.yaml	1
config/agent/components/hipbmdp_actor.yaml	6
config/agent/components/hipbmdp_critic.yaml	5
config/agent/components/hipbmdp_encoder.yaml	6
config/agent/components/hipbmdp_gradnorm.yaml	1
config/agent/components/hipbmdp_mask.yaml	3
config/agent/components/hipbmdp_multitask.yaml	33
config/agent/components/hipbmdp_task_to_encoder_cluster.yaml	1
config/agent/components/hipbmdp_transition_model.yaml	6
config/agent/components/metaworld_actor.yaml	6
config/agent/components/metaworld_critic.yaml	5
config/agent/components/metaworld_encoder.yaml	70
config/agent/components/metaworld_gradnorm.yaml	1
config/agent/components/metaworld_mask.yaml	3
config/agent/components/metaworld_multitask.yaml	30
config/agent/components/metaworld_task_to_encoder_cluster.yaml	22
config/agent/components/metaworld_transition_model.yaml	6
config/agent/deepmdp.yaml	32
config/agent/distral.yaml	42
config/agent/gradnorm_deepmdp.yaml	38
config/agent/gradnorm_sac.yaml	24
config/agent/gradnorm_state_deepmdp.yaml	38
config/agent/gradnorm_state_sac.yaml	24
config/agent/hipbmdp.yaml	32
config/agent/optimizers/hipbmdp_actor.yaml	3
config/agent/optimizers/hipbmdp_alpha.yaml	3
config/agent/optimizers/hipbmdp_critic.yaml	3
config/agent/optimizers/hipbmdp_decoder.yaml	4
config/agent/optimizers/hipbmdp_encoder.yaml	3
config/agent/optimizers/metaworld_actor.yaml	3
config/agent/optimizers/metaworld_alpha.yaml	3
config/agent/optimizers/metaworld_critic.yaml	3
config/agent/optimizers/metaworld_decoder.yaml	4
config/agent/optimizers/metaworld_encoder.yaml	3
config/agent/pcgrad_deepmdp.yaml	35
config/agent/pcgrad_sac.yaml	21
config/agent/pcgrad_sac_ae.yaml	27
config/agent/pcgrad_state_deepmdp.yaml	35
config/agent/pcgrad_state_sac.yaml	21
config/agent/sac.yaml	18
config/agent/sac_ae.yaml	24
config/agent/state_deepmdp.yaml	32
config/agent/state_sac.yaml	18
config/config.yaml	23
config/env/dmcontrol-finger-spin-distribution-v0.yaml	33
config/env/metaworld-ml1.yaml	8
config/env/metaworld-mt1.yaml	8
config/env/metaworld-mt10.yaml	22
config/env/metaworld-mt50.yaml	60
config/experiment/hipbmdp.yaml	18
config/experiment/metaworld.yaml	18
config/experiment/mtrl.yaml	18
config/logbook/mtrl.yaml	4
config/logger/mtrl.yaml	3
config/metrics/all.yaml	74
config/metrics/hipbmdp.yaml	71
config/metrics/metaworld.yaml	45
config/replay_buffer/mtrl.yaml	5
config/setup/hipbmdp.yaml	16
config/setup/metaworld.yaml	16
docs_src/source/conf.py	22
docs_src/source/pages/bib/refs.bib	67
main.py	10
mtrl/__init__.py	1
mtrl/agent/__init__.py	1
mtrl/agent/abstract.py	214
mtrl/agent/components/__init__.py	1
mtrl/agent/components/actor.py	284
mtrl/agent/components/base.py	9
mtrl/agent/components/critic.py	311
mtrl/agent/components/decoder.py	55
mtrl/agent/components/encoder.py	234
mtrl/agent/components/hipbmdp_theta.py	99
mtrl/agent/components/moe_layer.py	226
mtrl/agent/components/reward_decoder.py	20
mtrl/agent/components/soft_modularization.py	113
mtrl/agent/components/task_encoder.py	61
mtrl/agent/components/transition_model.py	109
mtrl/agent/deepmdp.py	225
mtrl/agent/distral.py	434
mtrl/agent/ds/__init__.py	1
mtrl/agent/ds/mt_obs.py	10
mtrl/agent/ds/task_info.py	11
mtrl/agent/grad_manipulation.py	98
mtrl/agent/gradnorm.py	123
mtrl/agent/hipbmdp.py	153
mtrl/agent/pcgrad.py	128
mtrl/agent/sac.py	468
mtrl/agent/sac_ae.py	133
mtrl/agent/utils.py	106
mtrl/agent/wrapper.py	94
mtrl/app/__init__.py	1
mtrl/app/run.py	22
mtrl/env/__init__.py	23
mtrl/env/builder.py	64
mtrl/env/types.py	9
mtrl/env/vec_env.py	88
mtrl/experiment/__init__.py	1
mtrl/experiment/dmcontrol.py	58
mtrl/experiment/experiment.py	143
mtrl/experiment/metaworld.py	155
mtrl/experiment/multitask.py	189
mtrl/experiment/utils.py	20
mtrl/logger.py	161
mtrl/replay_buffer.py	236
mtrl/utils/__init__.py	1
mtrl/utils/checkpointable.py	9
mtrl/utils/config.py	76
mtrl/utils/types.py	27
mtrl/utils/utils.py	41
mtrl/utils/video.py	28
noxfile.py	29
setup.cfg	64
setup.py	54
towncrier.toml	33