JAX EvoGym

Design voxel robots and simulate them in JAX. Run batched rollouts, build custom worlds, and render results.

Designer

A browser-based voxel editor for building EvoGym worlds. Import, validate, and export canonical JSON. Choose materials from the palette, with keyboard shortcuts for frequent actions.

Core Runtime

Object-separated JAX simulation with JIT-compiled physics and vmap-batched rollouts. Architecture docs cover compilation, collision, deformation, and the render model.

Canonical JSON

Designer import and export stay aligned with the EvoGym world format. Indices, types, and neighbor adjacency match EvoWorld.to_json_dict() exactly.

Static Deployment

The docs site and designer are pure frontend assets. Build once with Astro and host anywhere that serves static files.