libtpu 0.0.21.1__cp314-cp314t-manylinux_2_31_x86_64.whl

libtpu/__init__.py

@@ -0,0 +1,32 @@
+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+"""Wrapper for the libtpu library."""
+from importlib import metadata
+import os
+
+__version__ = metadata.version(__name__)
+
+
+def get_library_path() -> str:
+  return os.path.join(os.path.dirname(__file__), 'libtpu.so')
+
+
+def configure_library_path():
+  if not os.environ.get('TPU_LIBRARY_PATH'):
+    os.environ['TPU_LIBRARY_PATH'] = get_library_path()
+
+
+# on import of libtpu set the library path if not already configured.
+configure_library_path()

libtpu-0.0.21.1.dist-info/METADATA

@@ -0,0 +1,37 @@
+Metadata-Version: 2.4
+Name: libtpu
+Version: 0.0.21.1
+Summary: Google Cloud TPU runtime library.
+Author: Google, Inc.
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Dynamic: author
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: requires-python
+Dynamic: summary
+
+# 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](https://cloud.google.com/tpu).
+
+********************************************************************************
+
+## 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.

libtpu-0.0.21.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+libtpu/__init__.py,sha256=SoGncWaQwZYo1QEXcodPH75Y4C7El2NeXtDnDMAyL4M,1131
+libtpu/libtpu.so,sha256=fAB1hfNz8xYVhfsbSxvrDvPDkzXwlZBJx3u5rztERBc,376021592
+libtpu/sdk.so,sha256=RGJGSBFtQ2FF2rmfWbgDA1thLdEPdl9eXmgM-7AYNDQ,3595568
+libtpu-0.0.21.1.dist-info/METADATA,sha256=sN4PmqVaWrRDHbP0sXZEFBvejDbQgvN5pgFnP85w3l8,1123
+libtpu-0.0.21.1.dist-info/WHEEL,sha256=1DFW7SpqQQppRXWg2VpV00TC-vciwfOzgBAlBcviWiY,114
+libtpu-0.0.21.1.dist-info/top_level.txt,sha256=osLnDQBl5z7VhfpbHzbmqr5CNtGhnkU3Ue2zh1MfUKQ,7
+libtpu-0.0.21.1.dist-info/RECORD,,

libtpu-0.0.21.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: false
+Tag: cp314-cp314t-manylinux_2_31_x86_64
+

libtpu-0.0.21.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+libtpu