PJRT - Uniform Device API

PJRT C API は、ML エコシステムに追加する一元化された Device API です。長期的なビジョンは、（1）フレームワーク（TF、JAX など）が PJRT を呼び出すことです。PJRT には、フレームワークに対して不透明なデバイス固有の実装があります。また、（2）各デバイスは、PJRT プラグインとして PJRT API を実装することに重点を置いています。これは、フレームワークに対して不透明な場合があります。

通信チャネル

• 問題や機能リクエストは OpenXLA/xla リポジトリで報告できます。
• PJRT に関する質問は、OpenXLA Discord で質問できます。

リソース

• PJRT C API ヘッダー
• PJRT C API の変更ログ
• PJRT 統合ガイド
• PJRT 設計ドキュメント
• PJRT API ABI のバージョニングと互換性
• PJRT プラグイン メカニズムの設計書
• OpenXLA/IREE PJRT プラグインの実装