All APIs and reference documents

gcloud CLI

Command-Line Reference

Describes the Google Cloud CLI that supports Cloud TPU.

Frameworks (TensorFlow, PyTorch, JAX)

TensorFlow Reference

Links to the TensorFlow API documentation.
Available TensorFlow Ops

Lists the TensorFlow operations available on Cloud TPU.
Supported TPUs and TPU software versions

Lists the supported Cloud TPU configurations and supported framework (TensorFlow, PyTorch, JAX) software versions.
JAX AI stack

Describes the JAX AI stack.
PyTorch on TPU

TorchTPU is a native PyTorch backend for Cloud TPUs, built with the PyTorch team and integrated with the ecosystem of projects like vLLM-TorchTPU, SGLang-TorchTPU, and TorchTitan-TorchTPU.

TPU API

TPU API JSON Reference

Describes the RESTful services supported by the Cloud TPU API.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.