Skip to main content

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.45

Compatibility

  • Python Compatibility: The libtpu SDK is now compatible with the Python versions 3.11, 3.12, 3.13, 3.14 and 3.14-ft where ft corresponds to free threaded python variant. We have re-enabled support Python 3.11 to keep it compatible with TorchTPU.

Version 0.0.43

Compatibility

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

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.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

libtpu-0.0.45-cp314-cp314t-manylinux_2_31_x86_64.whl (215.9 MB view details)

Uploaded CPython 3.14tmanylinux: glibc 2.31+ x86-64

libtpu-0.0.45-cp314-cp314-manylinux_2_31_x86_64.whl (215.9 MB view details)

Uploaded CPython 3.14manylinux: glibc 2.31+ x86-64

libtpu-0.0.45-cp313-cp313-manylinux_2_31_x86_64.whl (215.9 MB view details)

Uploaded CPython 3.13manylinux: glibc 2.31+ x86-64

libtpu-0.0.45-cp312-cp312-manylinux_2_31_x86_64.whl (215.9 MB view details)

Uploaded CPython 3.12manylinux: glibc 2.31+ x86-64

libtpu-0.0.45-cp311-cp311-manylinux_2_31_x86_64.whl (215.9 MB view details)

Uploaded CPython 3.11manylinux: glibc 2.31+ x86-64

File details

Details for the file libtpu-0.0.45-cp314-cp314t-manylinux_2_31_x86_64.whl.

File metadata

File hashes

Hashes for libtpu-0.0.45-cp314-cp314t-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 0be7abed15eb561c604b862be5b0d85722f1aea56cc7c7882d7b0021853baa29
MD5 f0c9ff29c9165b309b94361ffdf8591e
BLAKE2b-256 52022c9a327dbdb4f9a225660ba43778c35aea959151e37fbd43fb155d18d5ce

See more details on using hashes here.

File details

Details for the file libtpu-0.0.45-cp314-cp314-manylinux_2_31_x86_64.whl.

File metadata

File hashes

Hashes for libtpu-0.0.45-cp314-cp314-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 a108aa9f413d09883e8e755c50df780dbb74b46965d93def15980df6774ea3b6
MD5 f765647dc6528acb6b53bdd3f419648f
BLAKE2b-256 689ec99345f52a86a4a5f72596a23a51fb3ba9ce14d511b28595fc019b5f7a36

See more details on using hashes here.

File details

Details for the file libtpu-0.0.45-cp313-cp313-manylinux_2_31_x86_64.whl.

File metadata

File hashes

Hashes for libtpu-0.0.45-cp313-cp313-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 dfc78556dc849fa776a32089334aad8042acbf72e6eb8e2a406ce4d140521740
MD5 c3a6d1651f444342d7ba6a562c495d25
BLAKE2b-256 7dd4f68b9174f455ccd2192a53714f66a0684324fdec119b9dd1e70b6ebe0b8e

See more details on using hashes here.

File details

Details for the file libtpu-0.0.45-cp312-cp312-manylinux_2_31_x86_64.whl.

File metadata

File hashes

Hashes for libtpu-0.0.45-cp312-cp312-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 9b6740915a1e2c802c3249d92608cbedf100036ce9c540100cdac063b94d22d9
MD5 5bbf7a7f48c49f67162e123f1bb24d75
BLAKE2b-256 b58d3e2d1708aa30f9382f41fa77e374e5eba071ec2bf181a6010c4d447833d1

See more details on using hashes here.

File details

Details for the file libtpu-0.0.45-cp311-cp311-manylinux_2_31_x86_64.whl.

File metadata

File hashes

Hashes for libtpu-0.0.45-cp311-cp311-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 1c0abca3497eee62278e199477d010f5ea51a32ff3ae3f66df82d0e2286589d3
MD5 5f602d7fb7d3c50314e42f1653a246b0
BLAKE2b-256 5f3c1de49cf114815067bb434bc6ac11e87ef4ebf834fe76e045ac5ee4cbae95

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page