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-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-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-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-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-cp314-cp314t-manylinux_2_31_x86_64.whl.

File metadata

File hashes

Hashes for libtpu-0.0.44-cp314-cp314t-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 bcf669216b54ea15dec92b4b459ac979b4312b8e3980539e073c61fe46121ecd
MD5 8cb73a72a707082a62173a374b5d735d
BLAKE2b-256 800f8adb8782e33608d7d1e38a19ab01e3ae234ddf58bf00abc3d71b4a8ee6a6

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for libtpu-0.0.44-cp314-cp314-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 fca31a295c503ac96c6275e67aa0dbe07c932e955c1c7972e05ea9d18ca80e5e
MD5 ae546befe80bb9cb6bb7ce37c0668156
BLAKE2b-256 0243f0c43ae31382de066743808d2295a09759ee6f3352ff6fd14013e7cb4ccb

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for libtpu-0.0.44-cp313-cp313-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 57d9a57bb4bea5b4d72ba5d65601599af6a032f005c1f82edb0fdf86beb63c97
MD5 8a378154b3c03ea84b3a9d86538a8b78
BLAKE2b-256 26cb08869affa6cc1b0bde1af9be9d63a4d7e29171b0eb112afc112e3d564233

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for libtpu-0.0.44-cp312-cp312-manylinux_2_31_x86_64.whl
Algorithm Hash digest
SHA256 ed5d909c4aeabfb57ec5198099f127c9702a3a3e9e142f76ddc4ee317f1d7022
MD5 e7990ee35af6bf0d9af62c78c8b06ad9
BLAKE2b-256 0c8ab6763d1d49c2da4892eddb9604d6d95fc9e1144e2c5dad54eaa4b77d5a83

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