libtpu 0.0.38


pip install libtpu

  Latest version

Released: Mar 19, 2026

Project Links

Meta
Author: Google, Inc.
Requires Python: >=3.11

Classifiers

What is libtpu?

libtpu is the core library that enables machine learning frameworks like JAX, PyTorch, and TensorFlow to execute models on Google Cloud TPUs. It provides core functionality for compilation, inter-chip communication (ICI), and runtime execution.

libtpu also includes a set of SDK primitives for direct TPU interaction and deployment.

Learn more about Cloud TPUs at Google Cloud TPUs.


Version 0.0.21.1

Compatibility

  • JAX Compatibility: libtpu supports JAX 0.7.1 or newer.

  • Python Compatibility: The libtpu SDK is now compatible with the Python versions 3.11, 3.12, 3.13, 3.13-ft, 3.14 and 3.14-ft where ft corresponds to free threaded python variant.

No dependencies