Building and training ML models with Vertex AI