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