libtpu 0.0.25


pip install libtpu

  Latest version

Released: Oct 20, 2025

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