torch
torchaudio
torchvision
numpy
wandb
gpustat
timm
einops
psutil
triton
ista-daslab-optimizers-cuda
