tqdm
numpy
torch>=2.2.0
transformers>=4.43.0
datasets
einops>=0.8.0