torch==2.1.1
torchvision==0.16.1
torchaudio==2.1.1
transformers
datasets
datasets[vision]
datasets[audio]
diffusers[torch]
wandb
jupyterlab
pandas
scipy
pillow
seaborn
hydra-core
matplotlib
ray[all]
knockknock
tqdm
git+https://github.com/RobustBench/robustbench.git
wilds
-e .