torch==2.5.1
numpy==1.22.0
einops==0.8.0
transformers==4.47.1
matplotlib==3.8.4
tqdm==4.67.1
pytest==8.3.4 