TF распределенный API в Beam

Question

Я тестирую zscore из tenzflowflow_transform, и мы добавим больше вариантов нормализации scikit. Я рассматриваю вопрос о том, писать ли код в TFT, который использует луч уменьшить или реализовать его непосредственно в API TF.

Можно ли получить рабочие узлы из конвейера лучей, чтобы я мог посеять, что TF распространил ClusterSpec с рабочими узлами?