Kmonte/add tb for glt#603

Draft

kmontemayor2-sc wants to merge 53 commits into

mainfrom

kmonte/add-tb-for-glt

Commits on Apr 28, 2026

Update generated protobuf artifacts
kmontemayor
committed
Enable Vertex AI TensorBoard for trainer jobs
kmontemayor
committed

Commits on Apr 29, 2026

test(tensorboard): add failing tests for TensorBoardWriter class

kmontemayor
and
claude
committed
test(tensorboard): cover close idempotency on no-op writer

kmontemayor
and
claude
committed
refactor(tensorboard): replace function API with TensorBoardWriter class

kmontemayor
and
claude
committed
refactor(v1-trainer): remove dead tensorboard writer plumbing

kmontemayor
and
claude
committed
refactor(examples): migrate homogeneous_training.py to TensorBoardWriter

kmontemayor
and
claude
committed
refactor(examples): migrate heterogeneous_training.py to TensorBoardWriter

kmontemayor
and
claude
committed
refactor(examples): migrate graph_store/homogeneous_training.py to TensorBoardWriter

kmontemayor
and
claude
committed
refactor(examples): migrate graph_store/heterogeneous_training.py to TensorBoardWriter

kmontemayor
and
claude
committed
cleanup
kmontemayor
committed
chore(examples): drop shouldLogToTensorboard: true from task configs

kmontemayor
and
claude
committed
refactor(tensorboard): move TensorBoardWriter to gigl/utils

kmontemayor
and
claude
committed
refactor(examples): drop should_log_to_tensorboard gate

kmontemayor
and
claude
committed
refactor(trainer): always forward tensorboard_logs_uri when proto is set

kmontemayor
and
claude
committed

Commits on Apr 30, 2026

refactor(tensorboard): replace from_uri with from_env

kmontemayor
and
claude
committed
refactor(examples): drop tensorboard URI plumbing in homogeneous_training

kmontemayor
and
claude
committed
refactor(examples): drop tensorboard URI plumbing from training examples

kmontemayor
and
claude
committed

Commits on May 1, 2026

Update
kmontemayor
committed

Commits on May 4, 2026

proto: add TrainerConfig.tensorboard_experiment_name
kmontemayor
committed
validation: tensorboard_experiment_name requires tensorboard_resource_name

kmontemayor
and
claude
committed
vertex_ai: add tensorboard_experiment_name to VertexAiJobConfig

kmontemayor
and
claude
committed
vertex_ai: helper to ensure Experiment exists with backing TB

kmontemayor
and
claude
committed
vertex_ai: submit with experiment when tensorboard_experiment_name is set

kmontemayor
and
claude
committed
launcher: thread tensorboard_experiment_name through _build_job_config
kmontemayor
committed
launcher: thread tensorboard_experiment_name through launch entrypoints

kmontemayor
and
claude
committed
test(launcher): tighten experiment_name negative assertions to assertIsNone
kmontemayor
committed
trainer: forward TrainerConfig.tensorboard_experiment_name to launcher

kmontemayor
and
claude
committed
examples: demo TrainerConfig.tensorboardExperimentName
kmontemayor
committed
validation: tensorboard_experiment_name also requires tensorboard_logs_uri
kmontemayor
committed
Revert "validation: tensorboard_experiment_name also requires tensorboard_logs_uri"
kmontemayor
committed
vertex_ai: sanitize job_name for ExperimentRun ID, validate experiment_name
kmontemayor
committed
vertex_ai: drop experiment_run kwarg, let SDK auto-generate the run
kmontemayor
committed
tensorboard: stream events from chief rank, drop submit(experiment=) path
kmontemayor
committed

Commits on May 5, 2026

vertex_ai: always pass tensorboard= so VAI job page links to TB
kmontemayor
committed
tensorboard: emit unique run names so multi-job comparison shows two runs
kmontemayor
committed
examples: scope TensorBoardWriter to a try/finally block in all training entrypoints

kmontemayor
and
claude
committed
tools: add dev_submit_tb_smoke_job + tb_smoke_main for fast TB iteration
kmontemayor
committed
tools: relocate smoke launcher to gigl.utils.dev (tools/ is gitignored)
kmontemayor
committed
smoke: bump machine_type to n1-standard-16 (n1-standard-2 unsupported)
kmontemayor
committed
vertex_ai: log TensorBoard URLs at submit time
kmontemayor
committed
examples: drop personal experiment name from e2e CORA task config
kmontemayor
committed
dev: remove gigl/utils/dev/ smoke tooling
kmontemayor
committed
docs: remove in-flight branch plan from git
kmontemayor
committed
vertex_ai: collapse submit_kwargs dict back to direct kwargs
kmontemayor
committed
v1: revert all v1 trainer changes — out of scope for this PR
kmontemayor
committed
docs: add Vertex AI doc-link references to TB code paths
kmontemayor
committed
proto: move tensorboard_experiment_name to VertexAiResourceConfig
kmontemayor
committed
examples: update stale CORA task-config comment to point at the new proto
kmontemayor
committed
launcher: revert use_cuda + storage container_uri fixes — separate PR

kmontemayor
and
claude
committed
examples: drop try/finally around training loop, call close() at end

kmontemayor
and
claude
committed
tensorboard_writer: hoist aiplatform import to module top

kmontemayor
and
claude
committed
tests: drop pure-mock test files

kmontemayor
and
claude
committed