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