gymnasium[box2d]>=1.0.0
numpy>=1.24.0
matplotlib>=3.7.0
optuna>=3.0.0
swig>=4.4.1
pyyaml>=6.0

[all]
tensorflow>=2.18.0
torch>=2.11.0
torchvision
torchaudio
gymnasium[atari]>=1.0.0
gymnasium[mujoco]>=1.0.0
ale-py>=0.8
jax>=0.10.0
flax>=0.12.6
optax>=0.2.8

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

[box2d]
gymnasium[box2d]>=1.0.0

[cuda]
nvidia-cuda-nvcc-cu12
tensorflow[and-cuda]>=2.18.0

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

[jax]
flax>=0.12.6
jax>=0.10.0
optax>=0.2.8

[mujoco]
gymnasium[mujoco]>=1.0.0

[onnx]
onnx>=1.14
onnxruntime>=1.16
onnxscript>=0.1

[tensorboard]
torch>=2.11.0
tensorboard>=2.14

[tensorflow]
tensorflow>=2.18.0

[tensorflow-gpu]
tensorflow[and-cuda]>=2.18.0

[torch]
torch>=2.11.0
torchvision
torchaudio

[wandb]
wandb>=0.16
