├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── agents ├── a2c.py ├── agents.py └── lpg_agent.py ├── environments ├── environments.py ├── gridworld │ ├── configs.py │ ├── custom_mazes.py │ └── gridworld.py ├── gymnax │ └── configs.py ├── level_sampler.py └── rollout.py ├── experiments ├── logging.py └── parse_args.py ├── meta ├── meta.py └── train.py ├── models ├── agent.py ├── common.py ├── lpg.py └── optim.py ├── run_cpu.sh ├── run_gpu.sh ├── setup ├── docker │ ├── Dockerfile │ ├── build_cpu.sh │ └── build_gpu.sh ├── requirements-base.txt ├── requirements-cpu.txt └── requirements-gpu.txt ├── train.py └── util ├── __init__.py ├── data.py ├── jax.py └── metrics.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/README.md -------------------------------------------------------------------------------- /agents/a2c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/agents/a2c.py -------------------------------------------------------------------------------- /agents/agents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/agents/agents.py -------------------------------------------------------------------------------- /agents/lpg_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/agents/lpg_agent.py -------------------------------------------------------------------------------- /environments/environments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/environments/environments.py -------------------------------------------------------------------------------- /environments/gridworld/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/environments/gridworld/configs.py -------------------------------------------------------------------------------- /environments/gridworld/custom_mazes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/environments/gridworld/custom_mazes.py -------------------------------------------------------------------------------- /environments/gridworld/gridworld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/environments/gridworld/gridworld.py -------------------------------------------------------------------------------- /environments/gymnax/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/environments/gymnax/configs.py -------------------------------------------------------------------------------- /environments/level_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/environments/level_sampler.py -------------------------------------------------------------------------------- /environments/rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/environments/rollout.py -------------------------------------------------------------------------------- /experiments/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/experiments/logging.py -------------------------------------------------------------------------------- /experiments/parse_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/experiments/parse_args.py -------------------------------------------------------------------------------- /meta/meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/meta/meta.py -------------------------------------------------------------------------------- /meta/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/meta/train.py -------------------------------------------------------------------------------- /models/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/models/agent.py -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/models/common.py -------------------------------------------------------------------------------- /models/lpg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/models/lpg.py -------------------------------------------------------------------------------- /models/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/models/optim.py -------------------------------------------------------------------------------- /run_cpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/run_cpu.sh -------------------------------------------------------------------------------- /run_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/run_gpu.sh -------------------------------------------------------------------------------- /setup/docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/setup/docker/Dockerfile -------------------------------------------------------------------------------- /setup/docker/build_cpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/setup/docker/build_cpu.sh -------------------------------------------------------------------------------- /setup/docker/build_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/setup/docker/build_gpu.sh -------------------------------------------------------------------------------- /setup/requirements-base.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/setup/requirements-base.txt -------------------------------------------------------------------------------- /setup/requirements-cpu.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/setup/requirements-cpu.txt -------------------------------------------------------------------------------- /setup/requirements-gpu.txt: -------------------------------------------------------------------------------- 1 | jax[cuda12_pip]==0.4.13 2 | -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/train.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/util/__init__.py -------------------------------------------------------------------------------- /util/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/util/data.py -------------------------------------------------------------------------------- /util/jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/util/jax.py -------------------------------------------------------------------------------- /util/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EmptyJackson/groove/HEAD/util/metrics.py --------------------------------------------------------------------------------