Training an ML model with Kubeflow