torch
torchvision
pyyaml

[fancy feature]
tensorboard
