stable_baselines3<3.0,>=2.9.0
jax<0.12.0,>=0.4.24
flax
optax
tqdm
rich
tfp-nightly>=0.26.0.dev20250831

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