├── .github └── workflows │ └── gh-pages.yml ├── .gitignore ├── LICENSE ├── README.md ├── amago ├── __init__.py ├── agent.py ├── cli_utils.py ├── envs │ ├── __init__.py │ ├── amago_env.py │ ├── builtin │ │ ├── __init__.py │ │ ├── alchemy.py │ │ ├── ale_retro.py │ │ ├── babyai.py │ │ ├── gymnax_envs.py │ │ ├── half_cheetah_v4_vel.py │ │ ├── mazerunner.py │ │ ├── metaworld_ml.py │ │ ├── popgym_envs.py │ │ ├── procgen_envs.py │ │ ├── tmaze.py │ │ ├── toy_gym.py │ │ └── xland_minigrid.py │ ├── env_utils.py │ └── exploration.py ├── experiment.py ├── hindsight.py ├── loading.py ├── nets │ ├── __init__.py │ ├── actor_critic.py │ ├── cnn.py │ ├── ff.py │ ├── goal_embedders.py │ ├── policy_dists.py │ ├── traj_encoders.py │ ├── transformer.py │ ├── tstep_encoders.py │ └── utils.py └── utils.py ├── docs ├── Makefile ├── _static │ └── custom.css ├── api │ ├── amago.agent.rst │ ├── amago.cli_utils.rst │ ├── amago.envs.amago_env.rst │ ├── amago.envs.builtin.alchemy.rst │ ├── amago.envs.builtin.ale_retro.rst │ ├── amago.envs.builtin.babyai.rst │ ├── amago.envs.builtin.gymnax_envs.rst │ ├── amago.envs.builtin.half_cheetah_v4_vel.rst │ ├── amago.envs.builtin.mazerunner.rst │ ├── amago.envs.builtin.metaworld_ml.rst │ ├── amago.envs.builtin.popgym_envs.rst │ ├── amago.envs.builtin.procgen_envs.rst │ ├── amago.envs.builtin.rst │ ├── amago.envs.builtin.tmaze.rst │ ├── amago.envs.builtin.toy_gym.rst │ ├── amago.envs.builtin.xland_minigrid.rst │ ├── amago.envs.env_utils.rst │ ├── amago.envs.exploration.rst │ ├── amago.envs.rst │ ├── amago.experiment.rst │ ├── amago.hindsight.rst │ ├── amago.loading.rst │ ├── amago.nets.actor_critic.rst │ ├── amago.nets.cnn.rst │ ├── amago.nets.ff.rst │ ├── amago.nets.goal_embedders.rst │ ├── amago.nets.policy_dists.rst │ ├── amago.nets.rst │ ├── amago.nets.traj_encoders.rst │ ├── amago.nets.transformer.rst │ ├── amago.nets.tstep_encoders.rst │ ├── amago.nets.utils.rst │ ├── amago.rst │ └── amago.utils.rst ├── citation.rst ├── conf.py ├── examples │ ├── 00_meta_frozen_lake.rst │ ├── 01_basic_gym.rst │ ├── 02_gymnax.rst │ ├── 03_popgym_suite.rst │ ├── 04_tmaze.rst │ ├── 05_dark_key_door.rst │ ├── 06_alchemy.rst │ ├── 07_metaworld.rst │ ├── 08_ale.rst │ ├── 09_multitask_procgen.rst │ ├── 10_babyai.rst │ ├── 11_xland_minigrid.rst │ ├── 12_half_cheetah_vel.rst │ ├── 13_mazerunner_relabeling.rst │ ├── 14_d4rl.rst │ ├── MUJOCO_LOG.TXT │ └── index.rst ├── index.rst ├── installation.rst ├── make.bat ├── media │ ├── alchemy.png │ ├── amago_big_logo.png │ ├── amago_logo_2.png │ ├── amago_logo_3.png │ ├── amago_overview.png │ ├── coinrun2.png │ ├── context_length_diagram.png │ ├── d4rl.png │ ├── frozen_lake_success_rates_by_episode.png │ ├── gopher.png │ ├── gymnax_logo.png │ ├── half_cheetah.png │ ├── icrl_example.png │ ├── in_context_rl.png │ ├── mazerunner.png │ ├── metamon_icon.png │ ├── minigrid.png │ ├── multiamago.png │ ├── popgym.png │ ├── robot.png │ ├── sawyer2.png │ ├── tmaze.png │ ├── wandb_metrics.png │ ├── wandb_xaxes.png │ └── xland_rules.png └── tutorial │ ├── async.rst │ ├── configuration.rst │ ├── create_dataset.rst │ ├── create_experiment.rst │ ├── customization.rst │ ├── index.rst │ ├── setup_env.rst │ └── track_results.rst ├── examples ├── 00_meta_frozen_lake.py ├── 01_basic_gym.py ├── 02_gymnax.py ├── 03_popgym_suite.py ├── 04_tmaze.py ├── 05_dark_key_door.py ├── 06_alchemy.py ├── 07_metaworld.py ├── 08_ale.py ├── 09_multitask_procgen.py ├── 10_babyai.py ├── 11_xland_minigrid.py ├── 12_half_cheetah_vel.py ├── 13_mazerunner_relabeling.py ├── 14_d4rl.py └── experimental │ └── 15_maniskill.py └── setup.py /.github/workflows/gh-pages.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/.github/workflows/gh-pages.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/README.md -------------------------------------------------------------------------------- /amago/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/__init__.py -------------------------------------------------------------------------------- /amago/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/agent.py -------------------------------------------------------------------------------- /amago/cli_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/cli_utils.py -------------------------------------------------------------------------------- /amago/envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/__init__.py -------------------------------------------------------------------------------- /amago/envs/amago_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/amago_env.py -------------------------------------------------------------------------------- /amago/envs/builtin/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/__init__.py -------------------------------------------------------------------------------- /amago/envs/builtin/alchemy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/alchemy.py -------------------------------------------------------------------------------- /amago/envs/builtin/ale_retro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/ale_retro.py -------------------------------------------------------------------------------- /amago/envs/builtin/babyai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/babyai.py -------------------------------------------------------------------------------- /amago/envs/builtin/gymnax_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/gymnax_envs.py -------------------------------------------------------------------------------- /amago/envs/builtin/half_cheetah_v4_vel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/half_cheetah_v4_vel.py -------------------------------------------------------------------------------- /amago/envs/builtin/mazerunner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/mazerunner.py -------------------------------------------------------------------------------- /amago/envs/builtin/metaworld_ml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/metaworld_ml.py -------------------------------------------------------------------------------- /amago/envs/builtin/popgym_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/popgym_envs.py -------------------------------------------------------------------------------- /amago/envs/builtin/procgen_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/procgen_envs.py -------------------------------------------------------------------------------- /amago/envs/builtin/tmaze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/tmaze.py -------------------------------------------------------------------------------- /amago/envs/builtin/toy_gym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/toy_gym.py -------------------------------------------------------------------------------- /amago/envs/builtin/xland_minigrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/builtin/xland_minigrid.py -------------------------------------------------------------------------------- /amago/envs/env_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/env_utils.py -------------------------------------------------------------------------------- /amago/envs/exploration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/envs/exploration.py -------------------------------------------------------------------------------- /amago/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/experiment.py -------------------------------------------------------------------------------- /amago/hindsight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/hindsight.py -------------------------------------------------------------------------------- /amago/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/loading.py -------------------------------------------------------------------------------- /amago/nets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/nets/__init__.py -------------------------------------------------------------------------------- /amago/nets/actor_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/nets/actor_critic.py -------------------------------------------------------------------------------- /amago/nets/cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/nets/cnn.py -------------------------------------------------------------------------------- /amago/nets/ff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/nets/ff.py -------------------------------------------------------------------------------- /amago/nets/goal_embedders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/nets/goal_embedders.py -------------------------------------------------------------------------------- /amago/nets/policy_dists.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/nets/policy_dists.py -------------------------------------------------------------------------------- /amago/nets/traj_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/nets/traj_encoders.py -------------------------------------------------------------------------------- /amago/nets/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/nets/transformer.py -------------------------------------------------------------------------------- /amago/nets/tstep_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/nets/tstep_encoders.py -------------------------------------------------------------------------------- /amago/nets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/nets/utils.py -------------------------------------------------------------------------------- /amago/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/amago/utils.py -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_static/custom.css: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/api/amago.agent.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.agent.rst -------------------------------------------------------------------------------- /docs/api/amago.cli_utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.cli_utils.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.amago_env.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.amago_env.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.alchemy.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.alchemy.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.ale_retro.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.ale_retro.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.babyai.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.babyai.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.gymnax_envs.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.gymnax_envs.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.half_cheetah_v4_vel.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.half_cheetah_v4_vel.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.mazerunner.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.mazerunner.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.metaworld_ml.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.metaworld_ml.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.popgym_envs.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.popgym_envs.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.procgen_envs.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.procgen_envs.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.tmaze.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.tmaze.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.toy_gym.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.toy_gym.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.builtin.xland_minigrid.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.builtin.xland_minigrid.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.env_utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.env_utils.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.exploration.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.exploration.rst -------------------------------------------------------------------------------- /docs/api/amago.envs.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.envs.rst -------------------------------------------------------------------------------- /docs/api/amago.experiment.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.experiment.rst -------------------------------------------------------------------------------- /docs/api/amago.hindsight.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.hindsight.rst -------------------------------------------------------------------------------- /docs/api/amago.loading.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.loading.rst -------------------------------------------------------------------------------- /docs/api/amago.nets.actor_critic.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.nets.actor_critic.rst -------------------------------------------------------------------------------- /docs/api/amago.nets.cnn.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.nets.cnn.rst -------------------------------------------------------------------------------- /docs/api/amago.nets.ff.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.nets.ff.rst -------------------------------------------------------------------------------- /docs/api/amago.nets.goal_embedders.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.nets.goal_embedders.rst -------------------------------------------------------------------------------- /docs/api/amago.nets.policy_dists.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.nets.policy_dists.rst -------------------------------------------------------------------------------- /docs/api/amago.nets.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.nets.rst -------------------------------------------------------------------------------- /docs/api/amago.nets.traj_encoders.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.nets.traj_encoders.rst -------------------------------------------------------------------------------- /docs/api/amago.nets.transformer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.nets.transformer.rst -------------------------------------------------------------------------------- /docs/api/amago.nets.tstep_encoders.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.nets.tstep_encoders.rst -------------------------------------------------------------------------------- /docs/api/amago.nets.utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.nets.utils.rst -------------------------------------------------------------------------------- /docs/api/amago.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.rst -------------------------------------------------------------------------------- /docs/api/amago.utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/api/amago.utils.rst -------------------------------------------------------------------------------- /docs/citation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/citation.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/examples/00_meta_frozen_lake.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/00_meta_frozen_lake.rst -------------------------------------------------------------------------------- /docs/examples/01_basic_gym.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/01_basic_gym.rst -------------------------------------------------------------------------------- /docs/examples/02_gymnax.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/02_gymnax.rst -------------------------------------------------------------------------------- /docs/examples/03_popgym_suite.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/03_popgym_suite.rst -------------------------------------------------------------------------------- /docs/examples/04_tmaze.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/04_tmaze.rst -------------------------------------------------------------------------------- /docs/examples/05_dark_key_door.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/05_dark_key_door.rst -------------------------------------------------------------------------------- /docs/examples/06_alchemy.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/06_alchemy.rst -------------------------------------------------------------------------------- /docs/examples/07_metaworld.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/07_metaworld.rst -------------------------------------------------------------------------------- /docs/examples/08_ale.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/08_ale.rst -------------------------------------------------------------------------------- /docs/examples/09_multitask_procgen.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/09_multitask_procgen.rst -------------------------------------------------------------------------------- /docs/examples/10_babyai.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/10_babyai.rst -------------------------------------------------------------------------------- /docs/examples/11_xland_minigrid.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/11_xland_minigrid.rst -------------------------------------------------------------------------------- /docs/examples/12_half_cheetah_vel.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/12_half_cheetah_vel.rst -------------------------------------------------------------------------------- /docs/examples/13_mazerunner_relabeling.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/13_mazerunner_relabeling.rst -------------------------------------------------------------------------------- /docs/examples/14_d4rl.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/14_d4rl.rst -------------------------------------------------------------------------------- /docs/examples/MUJOCO_LOG.TXT: -------------------------------------------------------------------------------- 1 | Fri Dec 20 06:00:19 2024 2 | ERROR: could not initialize GLFW 3 | 4 | -------------------------------------------------------------------------------- /docs/examples/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/examples/index.rst -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/installation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/installation.rst -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/media/alchemy.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/alchemy.png -------------------------------------------------------------------------------- /docs/media/amago_big_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/amago_big_logo.png -------------------------------------------------------------------------------- /docs/media/amago_logo_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/amago_logo_2.png -------------------------------------------------------------------------------- /docs/media/amago_logo_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/amago_logo_3.png -------------------------------------------------------------------------------- /docs/media/amago_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/amago_overview.png -------------------------------------------------------------------------------- /docs/media/coinrun2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/coinrun2.png -------------------------------------------------------------------------------- /docs/media/context_length_diagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/context_length_diagram.png -------------------------------------------------------------------------------- /docs/media/d4rl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/d4rl.png -------------------------------------------------------------------------------- /docs/media/frozen_lake_success_rates_by_episode.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/frozen_lake_success_rates_by_episode.png -------------------------------------------------------------------------------- /docs/media/gopher.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/gopher.png -------------------------------------------------------------------------------- /docs/media/gymnax_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/gymnax_logo.png -------------------------------------------------------------------------------- /docs/media/half_cheetah.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/half_cheetah.png -------------------------------------------------------------------------------- /docs/media/icrl_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/icrl_example.png -------------------------------------------------------------------------------- /docs/media/in_context_rl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/in_context_rl.png -------------------------------------------------------------------------------- /docs/media/mazerunner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/mazerunner.png -------------------------------------------------------------------------------- /docs/media/metamon_icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/metamon_icon.png -------------------------------------------------------------------------------- /docs/media/minigrid.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/minigrid.png -------------------------------------------------------------------------------- /docs/media/multiamago.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/multiamago.png -------------------------------------------------------------------------------- /docs/media/popgym.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/popgym.png -------------------------------------------------------------------------------- /docs/media/robot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/robot.png -------------------------------------------------------------------------------- /docs/media/sawyer2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/sawyer2.png -------------------------------------------------------------------------------- /docs/media/tmaze.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/tmaze.png -------------------------------------------------------------------------------- /docs/media/wandb_metrics.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/wandb_metrics.png -------------------------------------------------------------------------------- /docs/media/wandb_xaxes.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/wandb_xaxes.png -------------------------------------------------------------------------------- /docs/media/xland_rules.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/media/xland_rules.png -------------------------------------------------------------------------------- /docs/tutorial/async.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/tutorial/async.rst -------------------------------------------------------------------------------- /docs/tutorial/configuration.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/tutorial/configuration.rst -------------------------------------------------------------------------------- /docs/tutorial/create_dataset.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/tutorial/create_dataset.rst -------------------------------------------------------------------------------- /docs/tutorial/create_experiment.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/tutorial/create_experiment.rst -------------------------------------------------------------------------------- /docs/tutorial/customization.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/tutorial/customization.rst -------------------------------------------------------------------------------- /docs/tutorial/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/tutorial/index.rst -------------------------------------------------------------------------------- /docs/tutorial/setup_env.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/tutorial/setup_env.rst -------------------------------------------------------------------------------- /docs/tutorial/track_results.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/docs/tutorial/track_results.rst -------------------------------------------------------------------------------- /examples/00_meta_frozen_lake.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/00_meta_frozen_lake.py -------------------------------------------------------------------------------- /examples/01_basic_gym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/01_basic_gym.py -------------------------------------------------------------------------------- /examples/02_gymnax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/02_gymnax.py -------------------------------------------------------------------------------- /examples/03_popgym_suite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/03_popgym_suite.py -------------------------------------------------------------------------------- /examples/04_tmaze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/04_tmaze.py -------------------------------------------------------------------------------- /examples/05_dark_key_door.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/05_dark_key_door.py -------------------------------------------------------------------------------- /examples/06_alchemy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/06_alchemy.py -------------------------------------------------------------------------------- /examples/07_metaworld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/07_metaworld.py -------------------------------------------------------------------------------- /examples/08_ale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/08_ale.py -------------------------------------------------------------------------------- /examples/09_multitask_procgen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/09_multitask_procgen.py -------------------------------------------------------------------------------- /examples/10_babyai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/10_babyai.py -------------------------------------------------------------------------------- /examples/11_xland_minigrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/11_xland_minigrid.py -------------------------------------------------------------------------------- /examples/12_half_cheetah_vel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/12_half_cheetah_vel.py -------------------------------------------------------------------------------- /examples/13_mazerunner_relabeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/13_mazerunner_relabeling.py -------------------------------------------------------------------------------- /examples/14_d4rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/14_d4rl.py -------------------------------------------------------------------------------- /examples/experimental/15_maniskill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/examples/experimental/15_maniskill.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/UT-Austin-RPL/amago/HEAD/setup.py --------------------------------------------------------------------------------