torch
torchvision
numpy
gym==0.23.1
wandb
tqdm
scipy
opencv-python
plotly
