How to implement a stateful mean encoder in TensorFlow Federated to be used in build_federated_averaging_process?

Question

In Tensorflow Federated (TFF), you can pass to the tff.learning.build_federated_averaging_process a broadcast_process and an aggregation_proc