Add job events for build and provisioning

GitOrigin-RevId: d32fdd105ebb818ef26d274a6e0fc04ebc8380d9

Author: Weiwei Yang

axlearn/cloud/gcp/runners/gke.py

@@ -640,6 +640,9 @@ def _execute(self):
                     metrics.record_job_wait_for_build(
                         cfg.name, time.perf_counter() - wait_build_start
                     )
+                    self._maybe_publish(
+                        cfg.name, msg="Cloud build finished", state=JobLifecycleState.STARTING
+                    )
                 except RuntimeError as e:
                     logging.error("Bundling failed: %s. Aborting the job.", e)
                     return
@@ -649,6 +652,9 @@ def _execute(self):
                     self._pre_provisioner.create_for(self._inner)
 
                 self._inner.execute()
+                self._maybe_publish(
+                    cfg.name, msg="Provisioning resources", state=JobLifecycleState.STARTING
+                )
             elif status == GKERunnerJob.Status.SUSPENDED:
                 # Job is in SUSPENDED state.
                 if suspended_since is None: