stable_baselines3<3.0,>=2.4.0
jax>=0.4.12
jaxlib
flax
optax
tqdm
rich
tensorflow_probability

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