TPU Training Utilities

October 24, 2025 ยท View on GitHub

This directory contains tools for setting up and running training/inference jobs on Google Cloud TPUs.

1. Environment Setup

  • Setup Guide
    • Configure TPU VMs
    • Set up environment variables
    • Install required dependencies

2. Running Stuff On TPUs

Workflow

  1. First, set up your TPU environment using the env-setup scripts
  2. Then, use the run scripts to run your training/inference jobs

Note: See individual READMEs for detailed usage instructions.