torch>=1.8
torchvision
torchaudio
transformers
timm
