wehub-resource-sync
1.7 KiB
1.7 KiB
Project: /_project.yaml Book: /_book.yaml
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 channels
- Issues and feature requests can be filed in the OpenXLA/xla repo.
- Questions regarding PJRT can be asked on the OpenXLA Discord.