wget>=3.2
tqdm>=4.66.4
numpy<=2.0.0
torch>=2.3.1
scikit-learn>=1.3.2
pytorch-lightning>=2.3.3
matplotlib>3.6.1
opacus>=1.4.1
seaborn>=0.13.2
torchmetrics>=1.4.1
pandas>=1.2
litdata==0.2.30
zstd==1.5.5.1
einops>=0.7.0
ema-pytorch==0.4.5
