Federated Learning Module