torch
optuna
pytorch-lightning
transformers