jax==0.4.30
pytorch_lightning==2.2.1
scikit_learn==1.5.0
torch==2.2.1
torchvision==0.17.1
tensorboard==2.17.0
notebook==7.2.1
jupyter==1.0.0
papermill
