torch
torchvision
accelerate
pyyaml
