torch
numpy
gymnasium[classic-control]>=1.3.0
pygame-ce>=2.1.3
stable-baselines3
tqdm
wandb[media]
imageio[ffmpeg]
opencv-python
rich
psutil
nvidia-ml-py

[ale]
gymnasium[atari]
ale-py

[all]
rich
psutil
nvidia-ml-py
ale-py
gymnasium[atari,box2d,classic-control,mujoco]

[atari]
gymnasium[atari]
ale-py

[box2d]
gymnasium[box2d]

[classic]
gymnasium[classic-control]

[cli]
rich
psutil
nvidia-ml-py

[dev]
ruff
mypy
black
build
twine
rich
psutil
nvidia-ml-py
pytest>=8.4.2

[mujoco]
gymnasium[mujoco]
