numpy
matplotlib
scipy
optuna
wandb
torch
