torch
einops
termcolor
pyyaml
pandas
numpy
torchvision
pillow
setuptools
tensorboard
