torch
pytorch_lightning
wandb
numpy
pandas
scikit-learn
torcheval
pathlib
