libtpu

PyPI page
Home page
Author: Google, Inc.
Summary: 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, and includes a set of SDK primitives for direct TPU interaction and deployment. Learn more about Cloud TPUs at https://cloud.google.com/tpu.
Latest version: 0.0.13


Downloads last day: 5,834
Downloads last week: 48,443
Downloads last month: 262,889