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.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.44.1-cp314-cp314t-manylinux_2_31_x86_64.whl (215.6 MB view details)

Uploaded CPython 3.14tmanylinux: glibc 2.31+ x86-64

libtpu-0.0.44.1-cp314-cp314-manylinux_2_31_x86_64.whl (215.6 MB view details)

Uploaded CPython 3.14manylinux: glibc 2.31+ x86-64

libtpu-0.0.44.1-cp313-cp313-manylinux_2_31_x86_64.whl (215.6 MB view details)

Uploaded CPython 3.13manylinux: glibc 2.31+ x86-64

libtpu-0.0.44.1-cp312-cp312-manylinux_2_31_x86_64.whl (215.6 MB view details)

Uploaded CPython 3.12manylinux: glibc 2.31+ x86-64

File details

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

File metadata

File hashes

Hashes for libtpu-0.0.44.1-cp314-cp314t-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 1a03f48c6e3fdbebaf3d685bfa226d0796db4c6a97a0b7cc4922438603ccef59
MD5 3ec44f2c9fbb5af112ddb03ff6e73f8b
BLAKE2b-256 7d5fe651523459466fe7ea5423dfdc1372eab1a4a7775adba53bad366867fe20

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for libtpu-0.0.44.1-cp314-cp314-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 a42e6ab09b9e736d94a0a8501662de8a3180feecb3ce17bbb96935a85b4daa58
MD5 1daf974330cf1dce3a2aaad54235d2f7
BLAKE2b-256 db8dd5eeeafc33d073b748c54c2c7c2e5a9dabe2176bf2c6d76839d3ed0672cc

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for libtpu-0.0.44.1-cp313-cp313-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 a9ddf82730f44e9a61b6a2e3ebbfc7a9881583f6d74f7805e84fa9610c9a0ac0
MD5 42ff9f0e22edb23d9613a081bfe837d0
BLAKE2b-256 1b2742e544ac8d4a072f03b8335395060559d70db6b6cc995fa0fe5df66ce0cb

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for libtpu-0.0.44.1-cp312-cp312-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 4a96b7c320f63267e6ce82f323c2d8fca68395170ac98db4eb29d8e3cca2edf9
MD5 7ed619be51c6e97ebc1d01ede161ccbf
BLAKE2b-256 6f66e57cdbcfaeb1c232b685ee78c15e37a5a7cb3889e3f604db6ca80c0717a5

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