torch>=2.0
torchvision>=0.15
torchmetrics>=0.11
tqdm
numpy
matplotlib

[full]
torch-pruning>=1.3
onnx
onnxruntime

[pruning]
torch-pruning>=1.3
