absl-py
etils[epath]
mujoco>=3.4.0
numpy
warp-lang>=1.11.0

[cpu]
jax

[cuda]
jax[cuda12]

[dev]
pre-commit
pytest
pytest-xdist
ruff
pygls<2.0.0,>=1.0.0
lsprotocol<2024.0.0,>=2023.0.1
mujoco>=3.4.1.dev0
warp-lang>=1.11.0.dev0
