PJRT - 統一 Device API

PJRT C API 是想新增至機器學習生態系統的統一 Device API,長期願景如下:(1) 架構 (TF、JAX 等) 會呼叫 PJRT,而 PJRT 具有針對裝置的實作項目,這些項目對架構而言是不可見的;(2) 每部裝置都會將 PJRT API 實作為 PJRT 外掛程式,而這些外掛程式對架構而言是不可見的。

通訊管道

資源