torch
pytorch_lightning
scikit-learn
protobuf
pandas
