All APIs and reference documents

gcloud CLI

Command-Line Reference

Describes the Google Cloud CLI that supports Cloud TPU.

Frameworks (JAX, PyTorch, TensorFlow)

JAX AI stack

Describes the JAX AI stack on Cloud TPU.
JAX Docs

Links to the JAX developer documentation.
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.
TorchTPU Docs

Links to the TorchTPU developer documentation.
TensorFlow Reference

Links to the TensorFlow API documentation.
Available TensorFlow Ops

Lists the TensorFlow operations available on Cloud TPU.

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.