scikit-learn==1.3.2
pandas==2.1.4
transformers==4.37.2
sentence-transformers==2.3.1
tensorboardX==2.6.1
accelerate==0.27.2
tqdm==4.66.2
pytest==7.4.3
scipy==1.12.0
seaborn==0.13.2
pillow==10.2.0
numpy==1.26.3
matplotlib==3.8.2
wandb==0.16.3
pytorch-metric-learning==2.4.1
torch_geometric==2.5.0
