numpy
torch
torchvision
matplotlib
seaborn
tdqm
opencv-python
imageio
scikit-learn
umap-learn
python-dotenv
gymnasium
minigrid
datasets
autoclip