lightning
torch
torchvision
optuna
tqdm
