openxla/openxla-pjrt-plugin openxla/openxla-pjrt-plugin

  • Stars
    star
    23
  • Rank 1,016,462 (Top 21 %)
  • Language
    C++
  • License
    Apache License 2.0
  • Created over 1 year ago
  • Updated 10 months ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
openxla/openxla-pjrt-plugin
github

openxla

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

PJRT plugin for interfacing the OpenXLA compiler to Jax, PyTorch/XLA and TensorFlow

More Repositories

1

xla

A machine learning compiler for GPUs, CPUs, and ML accelerators
C++
2,107
star
2

stablehlo

Backward compatible ML compute opset inspired by HLO/MHLO
MLIR
169
star
3

community

Stores documents and resources used by the OpenXLA developer community
68
star
4

shardy

MLIR-based partitioning system
C++
24
star
5

iree-comparative-benchmark

Python
9
star
6

triton-old

C++
4
star