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