stable_baselines3>=2.3.0
jax
jaxlib
flax
tqdm
rich
tensorflow_probability

[:python_version < "3.9.0"]
optax<0.1.8

[:python_version >= "3.9.0"]
optax

[tests]
pytest
pytest-cov
pytest-env
pytest-xdist
mypy
ruff>=0.3.1
black<25,>=24.2.0
