torch>=2.2.0
datasets>=2.18.0
tqdm>=4.66.0
numpy>=1.26.0