tensorflow>=2.18.0
gymnasium>=1.0.0
numpy>=1.24.0
matplotlib>=3.7.0
optuna>=3.0.0

[all]
torch>=2.0.0
torchvision
torchaudio
gymnasium[atari]>=1.0.0
gymnasium[mujoco]>=1.0.0
ale-py>=0.8

[atari]
gymnasium[atari]>=1.0.0
ale-py>=0.8

[box2d]
gymnasium[box2d]>=1.0.0

[cuda]
nvidia-cuda-nvcc-cu12

[dev]
pytest>=7.0
pytest-cov
black
ruff
mypy
twine>=5.0
torch>=2.0.0

[mujoco]
gymnasium[mujoco]>=1.0.0

[torch]
torch>=2.0.0
torchvision
torchaudio
