This allows run names to show up properly in Vertex AI/TB and local TB.
Testing
Ran MaxText through XPK and viewed managed/local TB
Ran MaxText from TPU VM and viewed managed/local TB
Next Steps
Update run name prefix in cloud-accelerator-diagnostics package to get slightly better UI
Add Vertex AI experiment name config option in MaxText. This is for uploading different runs to the same experiment from TPU VM. This option already exists on XPK
Notes
Include run name in the TensorBoard subdirectory.
This allows run names to show up properly in Vertex AI/TB and local TB.
Testing
Next Steps