Going over the tutorials for TFF (tensorflow-federated), it seems that performing federated averaging and gradient descent iterations using TFF is well understood and can accomplished easily. However, for other training scenarios such as decision tree training it is not clear to me whether such an implementation is readily available. In particular, does the TF-DF (tensorflow decision forest) integrates well with TFF. If so, does anyone have an example for implementing a regression tree training? Thanks.