numpy>=1.21.0
gymnasium>=0.28.0
tqdm
stable-baselines3

[cpu]
jax[cpu]

[dev]
ruff
mypy
pre-commit
isort
pytest

[gpu]
jax[cuda12]
