The styling need a little work but the math behind this federated learning diagram is working. Linear regression on each node and then weighted averaging on the "server".

From: https://beta.observablehq.com/@grantcuster/federated-learning-diagram