numpy>=1.17
scipy>=1.6
pygame-ce
matplotlib
seaborn
pandas
gymnasium
dill
docopt
pyyaml
tqdm

[default]
numba
optuna
ffmpeg-python
PyOpenGL
pyvirtualdisplay
gymnasium

[deploy]
sphinx
sphinx_rtd_theme

[jax_agents]
numba
optuna
ffmpeg-python
PyOpenGL
pyvirtualdisplay
gymnasium
jax[cpu]
chex
dm-haiku
optax
dm-reverb[tensorflow]==0.6.1
dm-tree
rlax

[torch_agents]
numba
optuna
ffmpeg-python
PyOpenGL
pyvirtualdisplay
gymnasium
torch>=1.6.0
opencv-python
gymnasium[accept-rom-license,atari]
ale-py>=0.8.0
stable-baselines3
