openxla

Here are 3 public repositories matching this topic...

preetham-s7 / TileLens

Open-source AI Accelerator (GPU/TPU) Hardware-Software Co-Design Toolkit & Roofline Memory Tile Visualizer

Updated Aug 21, 2026
Python

kredd2506 / OpenXLA-benchmark

Technical architecture analysis and proposed 3×3 benchmarking protocol for the OpenXLA compiler across TPU v6e, NVIDIA H200, and AMD MI300X. Companion site + paper (DRAFT v2, April 2026).

benchmarking gpu llvm tpu jax xla mlir openxla stablehlo pjrt ml-compilers

Updated Apr 18, 2026
CSS

pedronahum / pypjrt

Star

A dependency-free Python client for the PJRT C API and the XLA FFI. Run compiled ML programs on CPUs, GPUs and TPUs from pure Python.

python machine-learning compiler runtime gpu accelerator ffi cuda ctypes triton zero-dependency tpu jax xla dlpack openxla stablehlo pjrt

Updated Jul 23, 2026
Python

Add this topic to your repo

To associate your repository with the openxla topic, visit your repo's landing page and select "manage topics."

Learn more