├── .dockerignore ├── .github └── workflows │ └── tests.yaml ├── .gitignore ├── Dockerfile ├── LICENSE ├── README.md ├── baselines.yaml ├── director ├── Dockerfile ├── __init__.py ├── agent.py ├── behaviors.py ├── configs.yaml ├── expl.py ├── hierarchy.py ├── jaxagent.py ├── jaxutils.py ├── nets.py ├── ninjax.py ├── requirements.txt ├── run_docker.sh └── train.py ├── dreamerv3 ├── __init__.py ├── agent.py ├── configs.yaml ├── main.py └── rssm.py ├── embodied ├── __init__.py ├── core │ ├── __init__.py │ ├── base.py │ ├── chunk.py │ ├── clock.py │ ├── driver.py │ ├── limiters.py │ ├── random.py │ ├── replay.py │ ├── selectors.py │ ├── streams.py │ └── wrappers.py ├── envs │ ├── atari.py │ ├── bsuite.py │ ├── crafter.py │ ├── dmc.py │ ├── dmlab.py │ ├── dummy.py │ ├── from_dm.py │ ├── from_gym.py │ ├── loconav.py │ ├── loconav_quadruped.py │ ├── loconav_quadruped.xml │ ├── minecraft.py │ ├── minecraft_factor.py │ ├── minecraft_flat.py │ ├── minecraft_keyboard.py │ ├── pinpad.py │ └── procgen.py ├── jax │ ├── __init__.py │ ├── agent.py │ ├── heads.py │ ├── internal.py │ ├── nets.py │ ├── opt.py │ ├── outs.py │ ├── transform.py │ └── utils.py ├── perf │ ├── test_bandwidth.py │ ├── test_distr.py │ ├── test_driver.py │ └── test_replay.py ├── run │ ├── __init__.py │ ├── eval_only.py │ ├── parallel.py │ ├── pretrain.py │ ├── train.py │ └── train_eval.py └── tests │ ├── test_driver.py │ ├── test_layer_scan.py │ ├── test_parallel.py │ ├── test_replay.py │ ├── test_sampletree.py │ ├── test_train.py │ └── utils.py ├── entrypoint.sh ├── plot.py ├── ppo ├── __init__.py ├── agent.py ├── configs.yaml ├── main.py └── nets.py ├── pyproject.toml ├── requirements.txt └── setup.py /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/.dockerignore -------------------------------------------------------------------------------- /.github/workflows/tests.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/.github/workflows/tests.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/README.md -------------------------------------------------------------------------------- /baselines.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/baselines.yaml -------------------------------------------------------------------------------- /director/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/Dockerfile -------------------------------------------------------------------------------- /director/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/__init__.py -------------------------------------------------------------------------------- /director/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/agent.py -------------------------------------------------------------------------------- /director/behaviors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/behaviors.py -------------------------------------------------------------------------------- /director/configs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/configs.yaml -------------------------------------------------------------------------------- /director/expl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/expl.py -------------------------------------------------------------------------------- /director/hierarchy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/hierarchy.py -------------------------------------------------------------------------------- /director/jaxagent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/jaxagent.py -------------------------------------------------------------------------------- /director/jaxutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/jaxutils.py -------------------------------------------------------------------------------- /director/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/nets.py -------------------------------------------------------------------------------- /director/ninjax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/ninjax.py -------------------------------------------------------------------------------- /director/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/requirements.txt -------------------------------------------------------------------------------- /director/run_docker.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/run_docker.sh -------------------------------------------------------------------------------- /director/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/director/train.py -------------------------------------------------------------------------------- /dreamerv3/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dreamerv3/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/dreamerv3/agent.py -------------------------------------------------------------------------------- /dreamerv3/configs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/dreamerv3/configs.yaml -------------------------------------------------------------------------------- /dreamerv3/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/dreamerv3/main.py -------------------------------------------------------------------------------- /dreamerv3/rssm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/dreamerv3/rssm.py -------------------------------------------------------------------------------- /embodied/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/__init__.py -------------------------------------------------------------------------------- /embodied/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/core/__init__.py -------------------------------------------------------------------------------- /embodied/core/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/core/base.py -------------------------------------------------------------------------------- /embodied/core/chunk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/core/chunk.py -------------------------------------------------------------------------------- /embodied/core/clock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/core/clock.py -------------------------------------------------------------------------------- /embodied/core/driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/core/driver.py -------------------------------------------------------------------------------- /embodied/core/limiters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/core/limiters.py -------------------------------------------------------------------------------- /embodied/core/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/core/random.py -------------------------------------------------------------------------------- /embodied/core/replay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/core/replay.py -------------------------------------------------------------------------------- /embodied/core/selectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/core/selectors.py -------------------------------------------------------------------------------- /embodied/core/streams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/core/streams.py -------------------------------------------------------------------------------- /embodied/core/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/core/wrappers.py -------------------------------------------------------------------------------- /embodied/envs/atari.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/atari.py -------------------------------------------------------------------------------- /embodied/envs/bsuite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/bsuite.py -------------------------------------------------------------------------------- /embodied/envs/crafter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/crafter.py -------------------------------------------------------------------------------- /embodied/envs/dmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/dmc.py -------------------------------------------------------------------------------- /embodied/envs/dmlab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/dmlab.py -------------------------------------------------------------------------------- /embodied/envs/dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/dummy.py -------------------------------------------------------------------------------- /embodied/envs/from_dm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/from_dm.py -------------------------------------------------------------------------------- /embodied/envs/from_gym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/from_gym.py -------------------------------------------------------------------------------- /embodied/envs/loconav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/loconav.py -------------------------------------------------------------------------------- /embodied/envs/loconav_quadruped.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/loconav_quadruped.py -------------------------------------------------------------------------------- /embodied/envs/loconav_quadruped.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/loconav_quadruped.xml -------------------------------------------------------------------------------- /embodied/envs/minecraft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/minecraft.py -------------------------------------------------------------------------------- /embodied/envs/minecraft_factor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/minecraft_factor.py -------------------------------------------------------------------------------- /embodied/envs/minecraft_flat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/minecraft_flat.py -------------------------------------------------------------------------------- /embodied/envs/minecraft_keyboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/minecraft_keyboard.py -------------------------------------------------------------------------------- /embodied/envs/pinpad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/pinpad.py -------------------------------------------------------------------------------- /embodied/envs/procgen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/envs/procgen.py -------------------------------------------------------------------------------- /embodied/jax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/jax/__init__.py -------------------------------------------------------------------------------- /embodied/jax/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/jax/agent.py -------------------------------------------------------------------------------- /embodied/jax/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/jax/heads.py -------------------------------------------------------------------------------- /embodied/jax/internal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/jax/internal.py -------------------------------------------------------------------------------- /embodied/jax/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/jax/nets.py -------------------------------------------------------------------------------- /embodied/jax/opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/jax/opt.py -------------------------------------------------------------------------------- /embodied/jax/outs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/jax/outs.py -------------------------------------------------------------------------------- /embodied/jax/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/jax/transform.py -------------------------------------------------------------------------------- /embodied/jax/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/jax/utils.py -------------------------------------------------------------------------------- /embodied/perf/test_bandwidth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/perf/test_bandwidth.py -------------------------------------------------------------------------------- /embodied/perf/test_distr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/perf/test_distr.py -------------------------------------------------------------------------------- /embodied/perf/test_driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/perf/test_driver.py -------------------------------------------------------------------------------- /embodied/perf/test_replay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/perf/test_replay.py -------------------------------------------------------------------------------- /embodied/run/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/run/__init__.py -------------------------------------------------------------------------------- /embodied/run/eval_only.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/run/eval_only.py -------------------------------------------------------------------------------- /embodied/run/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/run/parallel.py -------------------------------------------------------------------------------- /embodied/run/pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/run/pretrain.py -------------------------------------------------------------------------------- /embodied/run/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/run/train.py -------------------------------------------------------------------------------- /embodied/run/train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/run/train_eval.py -------------------------------------------------------------------------------- /embodied/tests/test_driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/tests/test_driver.py -------------------------------------------------------------------------------- /embodied/tests/test_layer_scan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/tests/test_layer_scan.py -------------------------------------------------------------------------------- /embodied/tests/test_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/tests/test_parallel.py -------------------------------------------------------------------------------- /embodied/tests/test_replay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/tests/test_replay.py -------------------------------------------------------------------------------- /embodied/tests/test_sampletree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/tests/test_sampletree.py -------------------------------------------------------------------------------- /embodied/tests/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/tests/test_train.py -------------------------------------------------------------------------------- /embodied/tests/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/embodied/tests/utils.py -------------------------------------------------------------------------------- /entrypoint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/entrypoint.sh -------------------------------------------------------------------------------- /plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/plot.py -------------------------------------------------------------------------------- /ppo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ppo/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/ppo/agent.py -------------------------------------------------------------------------------- /ppo/configs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/ppo/configs.yaml -------------------------------------------------------------------------------- /ppo/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/ppo/main.py -------------------------------------------------------------------------------- /ppo/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/ppo/nets.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danijar/embodied/HEAD/setup.py --------------------------------------------------------------------------------