The distributed TensorFlow setup