torch>=2.0
torchvision
onnx>=1.15
onnxruntime<1.17,>=1.16
onnxconverter-common
onnx2torch
torch-pruning
pyyaml
numpy
xgboost
scikit-learn
pandas
tqdm

[dev]
pytest
build
twine
