numpy<2.0
torch
torchvision
matplotlib
seaborn
tqdm
opencv-python
imageio
scikit-learn
umap-learn
python-dotenv
gymnasium
minigrid
datasets
autoclip
huggingface_hub
safetensors
accelerate