https://singularityfeed.com/federated-learning-and-custom-aggregation-schemes/