├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── __init__.py ├── agents ├── __init__.py ├── jax_pure_reward.py ├── planet.py ├── pure_reward.py └── sv2p.py ├── bin ├── __init__.py ├── e2e_test.py ├── eval.py └── train_eval.py ├── configs ├── jax_pure_reward.gin ├── jax_pure_reward_cem_cheetah.gin ├── planet.gin ├── planet_cem_atari.gin ├── planet_cem_cheetah.gin ├── planet_cem_walker.gin ├── planet_mppi_cheetah.gin ├── planet_offline_eval.gin ├── pure_reward.gin ├── pure_reward_cem_atari.gin ├── pure_reward_cem_cheetah.gin ├── pure_reward_offline_eval.gin ├── rewnet.gin ├── rewnet_cem_atari.gin ├── rewnet_cem_cheetah.gin ├── rewnet_offline_eval.gin ├── sv2p.gin ├── sv2p_cem_atari.gin ├── sv2p_offline_eval.gin ├── sv2p_pycem_cheetah.gin ├── sv2p_tfcem_cheetah.gin └── tests │ ├── jax_pure_reward.gin │ ├── jax_pure_reward_cem_cheetah.gin │ ├── planet.gin │ ├── planet_cem_atari.gin │ ├── planet_cem_cheetah.gin │ ├── sv2p.gin │ ├── sv2p_cem_atari.gin │ └── sv2p_tfcem_cheetah.gin ├── imported_models ├── __init__.py ├── common.py ├── layers.py ├── reward_models.py ├── sv2p.py ├── sv2p_hparams.py ├── sv2p_tests.py └── tests_utils.py ├── intro.ipynb ├── loops ├── __init__.py └── train_eval.py ├── objectives ├── __init__.py └── objectives.py ├── planners ├── __init__.py └── planners.py ├── setup.py ├── simulate ├── __init__.py └── simulate.py ├── tasks ├── __init__.py └── tasks.py └── utils ├── __init__.py ├── nested.py ├── npz.py ├── visualization.py └── wrappers.py /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/__init__.py -------------------------------------------------------------------------------- /agents/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/agents/__init__.py -------------------------------------------------------------------------------- /agents/jax_pure_reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/agents/jax_pure_reward.py -------------------------------------------------------------------------------- /agents/planet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/agents/planet.py -------------------------------------------------------------------------------- /agents/pure_reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/agents/pure_reward.py -------------------------------------------------------------------------------- /agents/sv2p.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/agents/sv2p.py -------------------------------------------------------------------------------- /bin/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/bin/__init__.py -------------------------------------------------------------------------------- /bin/e2e_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/bin/e2e_test.py -------------------------------------------------------------------------------- /bin/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/bin/eval.py -------------------------------------------------------------------------------- /bin/train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/bin/train_eval.py -------------------------------------------------------------------------------- /configs/jax_pure_reward.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/jax_pure_reward.gin -------------------------------------------------------------------------------- /configs/jax_pure_reward_cem_cheetah.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/jax_pure_reward_cem_cheetah.gin -------------------------------------------------------------------------------- /configs/planet.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/planet.gin -------------------------------------------------------------------------------- /configs/planet_cem_atari.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/planet_cem_atari.gin -------------------------------------------------------------------------------- /configs/planet_cem_cheetah.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/planet_cem_cheetah.gin -------------------------------------------------------------------------------- /configs/planet_cem_walker.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/planet_cem_walker.gin -------------------------------------------------------------------------------- /configs/planet_mppi_cheetah.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/planet_mppi_cheetah.gin -------------------------------------------------------------------------------- /configs/planet_offline_eval.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/planet_offline_eval.gin -------------------------------------------------------------------------------- /configs/pure_reward.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/pure_reward.gin -------------------------------------------------------------------------------- /configs/pure_reward_cem_atari.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/pure_reward_cem_atari.gin -------------------------------------------------------------------------------- /configs/pure_reward_cem_cheetah.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/pure_reward_cem_cheetah.gin -------------------------------------------------------------------------------- /configs/pure_reward_offline_eval.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/pure_reward_offline_eval.gin -------------------------------------------------------------------------------- /configs/rewnet.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/rewnet.gin -------------------------------------------------------------------------------- /configs/rewnet_cem_atari.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/rewnet_cem_atari.gin -------------------------------------------------------------------------------- /configs/rewnet_cem_cheetah.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/rewnet_cem_cheetah.gin -------------------------------------------------------------------------------- /configs/rewnet_offline_eval.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/rewnet_offline_eval.gin -------------------------------------------------------------------------------- /configs/sv2p.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/sv2p.gin -------------------------------------------------------------------------------- /configs/sv2p_cem_atari.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/sv2p_cem_atari.gin -------------------------------------------------------------------------------- /configs/sv2p_offline_eval.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/sv2p_offline_eval.gin -------------------------------------------------------------------------------- /configs/sv2p_pycem_cheetah.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/sv2p_pycem_cheetah.gin -------------------------------------------------------------------------------- /configs/sv2p_tfcem_cheetah.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/sv2p_tfcem_cheetah.gin -------------------------------------------------------------------------------- /configs/tests/jax_pure_reward.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/tests/jax_pure_reward.gin -------------------------------------------------------------------------------- /configs/tests/jax_pure_reward_cem_cheetah.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/tests/jax_pure_reward_cem_cheetah.gin -------------------------------------------------------------------------------- /configs/tests/planet.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/tests/planet.gin -------------------------------------------------------------------------------- /configs/tests/planet_cem_atari.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/tests/planet_cem_atari.gin -------------------------------------------------------------------------------- /configs/tests/planet_cem_cheetah.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/tests/planet_cem_cheetah.gin -------------------------------------------------------------------------------- /configs/tests/sv2p.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/tests/sv2p.gin -------------------------------------------------------------------------------- /configs/tests/sv2p_cem_atari.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/tests/sv2p_cem_atari.gin -------------------------------------------------------------------------------- /configs/tests/sv2p_tfcem_cheetah.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/configs/tests/sv2p_tfcem_cheetah.gin -------------------------------------------------------------------------------- /imported_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/imported_models/__init__.py -------------------------------------------------------------------------------- /imported_models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/imported_models/common.py -------------------------------------------------------------------------------- /imported_models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/imported_models/layers.py -------------------------------------------------------------------------------- /imported_models/reward_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/imported_models/reward_models.py -------------------------------------------------------------------------------- /imported_models/sv2p.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/imported_models/sv2p.py -------------------------------------------------------------------------------- /imported_models/sv2p_hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/imported_models/sv2p_hparams.py -------------------------------------------------------------------------------- /imported_models/sv2p_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/imported_models/sv2p_tests.py -------------------------------------------------------------------------------- /imported_models/tests_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/imported_models/tests_utils.py -------------------------------------------------------------------------------- /intro.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/intro.ipynb -------------------------------------------------------------------------------- /loops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/loops/__init__.py -------------------------------------------------------------------------------- /loops/train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/loops/train_eval.py -------------------------------------------------------------------------------- /objectives/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/objectives/__init__.py -------------------------------------------------------------------------------- /objectives/objectives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/objectives/objectives.py -------------------------------------------------------------------------------- /planners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/planners/__init__.py -------------------------------------------------------------------------------- /planners/planners.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/planners/planners.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/setup.py -------------------------------------------------------------------------------- /simulate/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/simulate/__init__.py -------------------------------------------------------------------------------- /simulate/simulate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/simulate/simulate.py -------------------------------------------------------------------------------- /tasks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/tasks/__init__.py -------------------------------------------------------------------------------- /tasks/tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/tasks/tasks.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/nested.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/utils/nested.py -------------------------------------------------------------------------------- /utils/npz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/utils/npz.py -------------------------------------------------------------------------------- /utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/utils/visualization.py -------------------------------------------------------------------------------- /utils/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/world_models/HEAD/utils/wrappers.py --------------------------------------------------------------------------------