packaging
gpustat
ipdb
matplotlib
tensorboard
einops
matplotlib
numpy
pandas
PyYAML
torch==2.0 --extra-index-url https://download.pytorch.org/whl/cu118
torchvision --extra-index-url https://download.pytorch.org/whl/cu118
git+https://github.com/fra31/auto-attack
gin-config
ipykernel
lion-pytorch
tqdm
timm