numpy
pandas
lightning
scikit_learn
torch
torchmetrics
setuptools
properscoring
