prolif==2.0.1
pandas==1.5.3
numpy
rdkit
torch
PyYAML
scikit_learn
tqdm
wandb
