torch
flwr
pytorch-lightning
numpy
