numpy
scipy
pandas
matplotlib
sklearn
pyyaml>=5.1
torch
pytorch-lightning
tqdm
