GitHub repository: openxla/openxla-pjrt-plugin
Primary language: C++
Created: 2023-02-09T18:18:54Z
Description: PJRT plugin for interfacing the OpenXLA compiler to Jax, PyTorch/XLA and TensorFlow