torch
torchvision
einops
tensorboard
tensorboardX
tqdm
matplotlib
comet_ml
gymnasium[atari,accept-rom-license]
opencv-python
omegaconf
hydra-core
colorama