stable_baselines3~=1.6.0
jax
jaxlib
flax
optax
tqdm
rich
tensorflow_probability

[tests]
pytest
pytest-cov
pytest-env
pytest-xdist
pytype
flake8>=3.8
importlib-metadata~=4.13
flake8-bugbear
isort>=5.0
black
