torch
flwr
pytorch-lightning
numpy
scikit-learn
