numpy
packaging
torch
typing-extensions>=3.10

[onnx]
onnx

[test]
pytest
onnxruntime
torchvision
