Extend the training job payload with an option to select the device to use for training among the ones available on the platform.