├── .gitignore ├── README.md ├── docs ├── craftax.md ├── getting_started.md ├── index.md ├── level_sampler.md ├── maze.md ├── maze_dr.md ├── maze_paired.md ├── maze_plr.md ├── underspecified_env.md └── wrappers.md ├── examples ├── craftax │ ├── __init__.py │ ├── craftax_plr.py │ ├── craftax_wrappers.py │ └── mutators.py ├── gymnax │ └── gymnax_plr.py ├── maze_dr.py ├── maze_paired.py └── maze_plr.py ├── figures ├── Labyrinth2_299.gif ├── LabyrinthFlipped_299.gif ├── Labyrinth_299.gif ├── SixteenRooms2_299.gif ├── SixteenRooms_299.gif ├── StandardMaze2_299.gif ├── StandardMaze3_299.gif └── StandardMaze_299.gif ├── mkdocs.yml ├── pyproject.toml ├── src └── jaxued │ ├── .DS_Store │ ├── __init__.py │ ├── environments │ ├── __init__.py │ ├── gymnax │ │ ├── acrobot.py │ │ ├── cartpole.py │ │ └── pendulum.py │ ├── maze │ │ ├── __init__.py │ │ ├── env.py │ │ ├── env_editor.py │ │ ├── env_solved.py │ │ ├── level.py │ │ ├── renderer.py │ │ └── util.py │ └── underspecified_env.py │ ├── level_sampler.py │ ├── linen.py │ ├── utils.py │ └── wrappers │ ├── __init__.py │ ├── autoreplay.py │ └── autoreset.py ├── tests └── test_examples_kinda.py └── tox.ini /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/README.md -------------------------------------------------------------------------------- /docs/craftax.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/docs/craftax.md -------------------------------------------------------------------------------- /docs/getting_started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/docs/getting_started.md -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/level_sampler.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/docs/level_sampler.md -------------------------------------------------------------------------------- /docs/maze.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/docs/maze.md -------------------------------------------------------------------------------- /docs/maze_dr.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/docs/maze_dr.md -------------------------------------------------------------------------------- /docs/maze_paired.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/docs/maze_paired.md -------------------------------------------------------------------------------- /docs/maze_plr.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/docs/maze_plr.md -------------------------------------------------------------------------------- /docs/underspecified_env.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/docs/underspecified_env.md -------------------------------------------------------------------------------- /docs/wrappers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/docs/wrappers.md -------------------------------------------------------------------------------- /examples/craftax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/craftax/craftax_plr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/examples/craftax/craftax_plr.py -------------------------------------------------------------------------------- /examples/craftax/craftax_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/examples/craftax/craftax_wrappers.py -------------------------------------------------------------------------------- /examples/craftax/mutators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/examples/craftax/mutators.py -------------------------------------------------------------------------------- /examples/gymnax/gymnax_plr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/examples/gymnax/gymnax_plr.py -------------------------------------------------------------------------------- /examples/maze_dr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/examples/maze_dr.py -------------------------------------------------------------------------------- /examples/maze_paired.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/examples/maze_paired.py -------------------------------------------------------------------------------- /examples/maze_plr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/examples/maze_plr.py -------------------------------------------------------------------------------- /figures/Labyrinth2_299.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/figures/Labyrinth2_299.gif -------------------------------------------------------------------------------- /figures/LabyrinthFlipped_299.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/figures/LabyrinthFlipped_299.gif -------------------------------------------------------------------------------- /figures/Labyrinth_299.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/figures/Labyrinth_299.gif -------------------------------------------------------------------------------- /figures/SixteenRooms2_299.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/figures/SixteenRooms2_299.gif -------------------------------------------------------------------------------- /figures/SixteenRooms_299.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/figures/SixteenRooms_299.gif -------------------------------------------------------------------------------- /figures/StandardMaze2_299.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/figures/StandardMaze2_299.gif -------------------------------------------------------------------------------- /figures/StandardMaze3_299.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/figures/StandardMaze3_299.gif -------------------------------------------------------------------------------- /figures/StandardMaze_299.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/figures/StandardMaze_299.gif -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/pyproject.toml -------------------------------------------------------------------------------- /src/jaxued/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/.DS_Store -------------------------------------------------------------------------------- /src/jaxued/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/jaxued/environments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/environments/__init__.py -------------------------------------------------------------------------------- /src/jaxued/environments/gymnax/acrobot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/environments/gymnax/acrobot.py -------------------------------------------------------------------------------- /src/jaxued/environments/gymnax/cartpole.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/environments/gymnax/cartpole.py -------------------------------------------------------------------------------- /src/jaxued/environments/gymnax/pendulum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/environments/gymnax/pendulum.py -------------------------------------------------------------------------------- /src/jaxued/environments/maze/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/environments/maze/__init__.py -------------------------------------------------------------------------------- /src/jaxued/environments/maze/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/environments/maze/env.py -------------------------------------------------------------------------------- /src/jaxued/environments/maze/env_editor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/environments/maze/env_editor.py -------------------------------------------------------------------------------- /src/jaxued/environments/maze/env_solved.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/environments/maze/env_solved.py -------------------------------------------------------------------------------- /src/jaxued/environments/maze/level.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/environments/maze/level.py -------------------------------------------------------------------------------- /src/jaxued/environments/maze/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/environments/maze/renderer.py -------------------------------------------------------------------------------- /src/jaxued/environments/maze/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/environments/maze/util.py -------------------------------------------------------------------------------- /src/jaxued/environments/underspecified_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/environments/underspecified_env.py -------------------------------------------------------------------------------- /src/jaxued/level_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/level_sampler.py -------------------------------------------------------------------------------- /src/jaxued/linen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/linen.py -------------------------------------------------------------------------------- /src/jaxued/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/utils.py -------------------------------------------------------------------------------- /src/jaxued/wrappers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/wrappers/__init__.py -------------------------------------------------------------------------------- /src/jaxued/wrappers/autoreplay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/wrappers/autoreplay.py -------------------------------------------------------------------------------- /src/jaxued/wrappers/autoreset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/src/jaxued/wrappers/autoreset.py -------------------------------------------------------------------------------- /tests/test_examples_kinda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/tests/test_examples_kinda.py -------------------------------------------------------------------------------- /tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DramaCow/jaxued/HEAD/tox.ini --------------------------------------------------------------------------------