torch
huggingface_hub>=0.25
safetensors
numpy

[dev]
nbdev
torchvision
torch-pruning
onnx
fastcore

[test]
torchvision
torch-pruning
onnx
fastcore
