numpy>=1.21.2
onnxruntime

[hf]
huggingface_hub

[test]
pytest
scikit-learn

[train]
scikit-learn
joblib
skl2onnx
onnx
model2vec
mlflow
datasets
