numpy
gymnasium>=1.0.0
jax<0.8,>=0.5
jaxlib
optax
tfp-nightly[jax]
chex
tqdm
flax<0.12

[all]
rl_blox[dev,doc,examples,logging]

[dev]
pytest>=6.0
pre-commit
flake8
black
isort
pytest-cov

[doc]
sphinx
pydata-sphinx-theme
numpydoc

[examples]
gymnasium[classic-control,mujoco,toy-text]
matplotlib

[logging]
aim
orbax-checkpoint
