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.
Quick Links
1. Environment Setup
- Setup Guide
- Configure TPU VMs
- Set up environment variables
- Install required dependencies
2. Running Stuff On TPUs
- Training/Inference Guide
- Run training/inference jobs
- Monitor progress
- View logs
Workflow
- First, set up your TPU environment using the
env-setupscripts - Then, use the
runscripts to run your training/inference jobs
Note: See individual READMEs for detailed usage instructions.