datasets>=2.15
scikit-learn>=1.3.2
torch>=2.1
torchinfo>=1.8.0
torchvision>=0.16
tqdm>=4.66
transformers>=4.35
webdataset>=0.2.77
