psutil>=5.9
pyyaml>=6.0
rich>=13.0
numpy>=1.24
pandas>=2.0

[all]
docker>=7.0
torch>=2.0
torchvision>=0.15
nn-meter>=1.4
transformers>=4.0
accelerate>=0.20
scikit-learn>=1.0
hummingbird-ml>=0.4
joblib>=1.3
onnxruntime>=1.15

[docker]
docker>=7.0

[hf]
transformers>=4.0
accelerate>=0.20

[onnx]
onnxruntime>=1.15

[pytorch]
torch>=2.0
torchvision>=0.15
nn-meter>=1.4

[sklearn]
scikit-learn>=1.0
hummingbird-ml>=0.4
joblib>=1.3
