Google Cloud TPU runtime library.
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
supportsJAX
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 whereft
corresponds to free threaded python variant.
Oct 20, 2025
0.0.25
Oct 02, 2025
0.0.24
Oct 08, 2025
0.0.23.1
Sep 12, 2025
0.0.23
Sep 04, 2025
0.0.22
Aug 25, 2025
0.0.21.1
Aug 15, 2025
0.0.21
Sep 05, 2025
0.0.20.1
Aug 01, 2025
0.0.20
Jul 31, 2025
0.0.19.1
Jul 22, 2025
0.0.19
Jul 10, 2025
0.0.18
Jun 13, 2025
0.0.17
Jun 02, 2025
0.0.16
May 15, 2025
0.0.15
May 01, 2025
0.0.14
Apr 15, 2025
0.0.13
Apr 02, 2025
0.0.12
Jun 04, 2025
0.0.11.4
May 27, 2025
0.0.11.3
May 23, 2025
0.0.11.2
Mar 26, 2025
0.0.11.1
Mar 18, 2025
0.0.11
Feb 27, 2025
0.0.10.1
Feb 21, 2025
0.0.10
Feb 18, 2025
0.0.9
Feb 05, 2025
0.0.8
Feb 05, 2025
0.0.7.1
Feb 05, 2025
0.0.7
Feb 05, 2025
0.0.6
Feb 05, 2025
0.0.5
Feb 05, 2025
0.0.4
Feb 05, 2025
0.0.3
Feb 05, 2025
0.0.2
Feb 05, 2025
0.0.1
Wheel compatibility matrix
Platform | CPython 3.11 | CPython 3.12 | CPython 3.13 | CPython (additional flags: t) 3.13 | CPython (additional flags: t) 3.14 |
---|---|---|---|---|---|
manylinux_2_31_x86_64 |
Files in release
No dependencies