numpy>=1.18.0
scipy>=1.4.0
scikit-learn>=0.22.0
setuptools>=41.0.0
tqdm
surgeon_pytorch
safetensors
torch
torchvision
faiss-cpu