stable_baselines3>=2.3.0a1
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
black
