torch==2.0.1
accelerate==0.26.1
einops==0.6.0
matplotlib==3.8.2
numpy==1.24.4
pandas==2.1.4
scikit-learn==1.3.2
scipy==1.11.4
tqdm==4.66.1
transformers==4.36.2