torch>=1.13.0
numpy
rdkit
gdown
tqdm

[eval]
matplotlib
seaborn

[train]
lightning
wandb
