torch>=2.0
torchvision
numpy
scikit-learn
tensorflow-datasets
onnx
onnxruntime  # optional: for running/parity check