stablehlo
A functional language for machine learning. Inspired by Clojure, compiled to GPUs.
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