gpustat
wandb
torch==2.0.1
torchvision==0.15.2
numpy==1.24.3
torch_optimizer
adamp
pandas
tqdm
scipy
datasets
evaluate
pyhessian
scikit-learn