imageio
numpy
scipy
#torch
#torchvision
tqdm
wandb
prodigyopt
jupyter
scikit-learn
FCD==1.1 # to get the tensorflow version
PyTDC # for DRD2 computation