Skip to content

Add reproducible Cloud TPU v5e setup workflow - #116

Merged
charlesmartin14 merged 5 commits into
mainfrom
agent/add-v5e-tpu-setup-script
Aug 24, 2026
Merged

Add reproducible Cloud TPU v5e setup workflow#116
charlesmartin14 merged 5 commits into
mainfrom
agent/add-v5e-tpu-setup-script

Conversation

@charlesmartin14

@charlesmartin14 charlesmartin14 commented Aug 24, 2026

Copy link
Copy Markdown
Member

Summary

  • add baseline/nanogpt_one_head/setup_tpu_v5e.sh
  • capture the known-good PyTorch 2.6 / PyTorch-XLA 2.6 installation sequence
  • repair stale UNKNOWN-0.0.0 packaging artifacts and old pip/setuptools/wheel
  • persist TPU environment variables across SSH sessions without hard-coded home paths
  • support explicit ephemeral or persistent storage
  • verify raw XLA devices and rg-onehead-env --device auto
  • optionally prepare and run a small non-scientific AdamW TPU smoke benchmark
  • update TPU.md with Flex-Start provisioning, SSH, setup, diagnostics, and cleanup

Validation

  • bash -n baseline/nanogpt_one_head/setup_tpu_v5e.sh
  • verified --help output
  • verified the committed Git blob SHA matches the locally validated script
  • verified the utility and documentation satisfy the repository's no-home-path and no-wrapper-directory invariants
  • installation sequence is grounded in the successful v5litepod-4 session: torch 2.6.0+cu124, torch_xla 2.6.0, and four visible XLA devices

@charlesmartin14
charlesmartin14 merged commit 5b93a0a into main Aug 24, 2026
6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant