pjrt

Bringing StableHLO to Apple Silicon

A Rust Framework for Tracing, Automatic Differentiation, and Just-In-Time Compilation

a Haskell library and runtime for building, compiling, and executing machine learning programs targeting StableHLO

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).

Run JAX on MacBook Apple GPUs with pinned Metal/MPS profiles, jax-metal/jax-metallib compatibility checks, CPU fallback, and benchmarks.

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.

Tensor Logic as a StableHLO-emitting DSL for PJRT runtimes

不经过 libtpu/PJRT 的 TPU v4 最小用户态 runtime