stable_baselines3<3.0,>=2.8.0a0
jax<0.9.0,>=0.4.24
jaxlib
flax
optax
tqdm
rich
tfp-nightly>=0.26.0.dev20250831

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