optuna
torch
torchvision
pyyaml

[dev]
pytest
black
ruff
