Skip to content

Releases: google-pytorch/tpu-agent-suite

TPU agent suite v0.7.0-experimental

Pre-release

Choose a tag to compare

@learning-to-play learning-to-play released this 20 Oct 03:57

Initial Experimental Release of the TPU agent suite, released alongside torch_tpu at the PyTorch Conference 2026.

1. What is Included

  • Multi-Harness Native Support: Compiles canonical agent packages into native bundles for Claude Code and Antigravity (with Codex and Cursor coming soon).
  • Deterministic Execution: Uses explicit hooks, schemas, and script-enforced validation instead of unguided model behavior.
  • Orchestrator and Specialized Domain Agents:
    • tpu-agent: Orchestrator for the TPU agent suite.
    • tpu-model-onboarding: Onboards PyTorch models to verified TPU inference endpoints with automated hardware preflights and per-layer numerical parity checks.
    • tpu-diagnostics: Captures and reads XProf traces, tracing execution time back to source lines.
    • tpu-knowledge-base: Retrieval-augmented TPU fact-checking backed by ~600 catalog entries and strict citation contracts.
    • tpu-kernel-development: Generates, tests, and autotunes custom Pallas kernels.

2. Scope & Access