torch==2.4.0
torchvision==0.19.0

absl-py==2.1.0
einops==0.8.0
scikit-learn==1.5.2
tqdm==4.66.5

matplotlib==3.9.2
seaborn==0.13.2

datasets==3.1.0

