Back to Tensorflow

PJRT - Uniform Device API

third_party/xla/xla/pjrt/c/README.md

2.21.01.4 KB
Original Source

PJRT - Uniform Device API

PJRT C API is the uniform Device API that we want to add to the ML ecosystem. The long term vision is that: (1) frameworks (TF, JAX, etc.) will call PJRT, which has device-specific implementations that are opaque to the frameworks; (2) each device focuses on implementing PJRT APIs as PJRT plugins, which can be opaque to the frameworks.

Communication channel

Resources