torch>=1.13.1
torchvision>=0.14.1
numpy
matplotlib==3.5.2
tikzplotlib
pillow
PyYAML>=6.0
h5py
psutil
pynvml
scipy
scikit-learn
git+https://github.com/BorealisAI/advertorch.git
