PJRT - Uniform Device API

PJRT C API 是我们希望添加到机器学习生态系统中的统一设备 API。 我们的长期愿景是:(1) 框架(TF、JAX 等)将调用 PJRT,PJRT 具有对框架不透明的设备专用实现;(2) 每台设备都专注于将 PJRT API 实现为 PJRT 插件,这些插件对框架不透明。

通信通道

资源