Oven

libtpu0.0.23

Published 5 days ago

Google Cloud TPU runtime library.

pip install libtpu

Package Downloads

Weekly Downloads

Monthly Downloads

Authors

Requires Python

>=3.11

Dependencies

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.