Installation

Base package

pip install .

With visualization support (includes Pillow):

pip install ".[viz]"

For local development:

pip install -e ".[viz,dev]"

If you use uv:

uv pip install -e ".[viz,dev]"

JAX backend

jax-evogym works with any JAX backend. Install JAX separately for your target platform:

# CPU only
pip install jax

# CUDA (Linux)
pip install jax[cuda12]

See the JAX installation guide for full platform instructions.

Docs site and designer

The documentation site and browser-based designer are a separate Astro/Starlight frontend. They do not require JAX.

bun --cwd site install
bun --cwd site dev

Build the static site:

bun --cwd site build

The designer is served at /designer during local development and in the static build.