torch
torchaudio
torchvision
numpy
wandb
gpustat
timm
einops
psutil
