GitHub repository: ramiro050/openxla-pjrt-plugin
Primary language: C++
Description: PJRT plugin for interfacing the OpenXLA compiler to Jax, PyTorch/XLA and TensorFlow