Google has introduced an open-source TPU agent suite that uses natural language to automate TPU setup, diagnose bottlenecks and tune custom kernels. The suite is available in preview.
Connecting to coding tools
The agents connect directly to coding tools including Claude Code, Cursor and Google Antigravity. Google says the suite can automate TPU setup in minutes.
Native PyTorch support on TPUs
Google also highlighted TorchTPU, which it open-sourced last month. It enables PyTorch code and Hugging Face models to run natively on TPUs. Ready-to-use TPU backends are also available for the vLLM and SGLang inference engines.
