torch>=2.1.0
numpy
packaging
cloudpickle
tensordict>=0.4.0

[all]
ale-py
atari-py
dm_control
git
gym
gym[accept-rom-license]
gymnasium
h5py
huggingface_hub
hydra-core>=1.1
hydra-submitit-launcher
minari
moviepy
mujoco
pandas
pettingzoo>=1.24.1
pillow
pygame
pytest
pytest-instafail
pyyaml
requests
scikit-learn
scipy
tensorboard
torchsnapshot
torchvision
tqdm
vmas>=1.2.10
wandb

[atari]
gym
atari-py
ale-py
gym[accept-rom-license]
pygame

[checkpointing]
torchsnapshot

[dm_control]
dm_control

[gym_continuous]
gymnasium
mujoco

[marl]
vmas>=1.2.10
pettingzoo>=1.24.1

[offline-data]
huggingface_hub
minari
requests
tqdm
torchvision
scikit-learn
pandas
h5py
pillow

[rendering]
moviepy

[tests]
pytest
pyyaml
pytest-instafail
scipy

[utils]
tensorboard
wandb
tqdm
hydra-core>=1.1
hydra-submitit-launcher
git
