torch>=1.8
numpy
tqdm
matplotlib
pytorch_lightning
optuna

[:python_version >= "3.7"]
wandb
