torch==2.4.1
torchvision==0.19.1
torchaudio==2.4.1
lightning==2.4.0
matplotlib==3.9.2
numpy==2.1.1
scikit-learn==1.5.2
notebook==7.3.2