atari-py
opencv-python
plotly
torch
torchvision
kornia
tqdm
