torch
pytorch_lightning
wandb
numpy
pandas
scikit-learn
pathlib
