scikit-learn>=1.0.2
torch>=1.10.*
torchinfo>=1.6.5
torchvision>=0.11.*
Pillow>=7.1.2
tqdm>=4.63.*
sentencepiece>=0.1.96
sortedcontainers>=2.4.0

[azure]
azure-storage-blob>=12.13.0

[full]
ednaml[azure,metrics]

[metrics]
torchmetrics>=0.9.3

[transformers]
transformers>=4.21.*
