Distributed training and Hyperparameter tuning with TensorFlow on Vertex AI