torch
torchvision

[tune]
ray[tune]
optuna
