numpy==1.25.2
datasets==2.14.5
matplotlib==3.8.0
torch>=2.1.0
torchvision>=0.17.2
tqdm==4.66.1
transformers==4.40.2
wandb==0.15.12
hadamard_transform==0.1.3
scipy==1.12.0
seaborn==0.11.2