loguru
optuna
torch
torchinfo
