stable_baselines3<3.0,>=2.8.0a0
jax<0.7.0,>=0.4.24
jaxlib
flax
optax
tqdm
rich
tensorflow_probability

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