numpy~=1.26.4
torch~=2.3.1+cu121
matplotlib~=3.9.2
tqdm~=4.66.4
scipy~=1.14.0
scikit-learn~=1.5.1
pandas~=2.2.2
seaborn~=0.13.2
setuptools~=65.5.0
six~=1.16.0
torchvision~=0.18.1+cu121