torch>=2.0
numpy

[bench]
scikit-learn
pandas
matplotlib

[bench-all]
scikit-learn
pandas
matplotlib
torchvision
datasets
transformers>=4.40
peft>=0.10

[bench-vision]
torchvision
datasets

[telemetry]
opentelemetry-sdk>=1.20
opentelemetry-exporter-otlp>=1.20

[text]
transformers>=4.40
peft>=0.10
