PJRT C API 是我们希望添加到机器学习生态系统中的统一设备 API。 我们的长期愿景是:(1) 框架(TF、JAX 等)将调用 PJRT,PJRT 具有对框架不透明的设备专用实现;(2) 每台设备都专注于将 PJRT API 实现为 PJRT 插件,这些插件对框架不透明。
通信通道
- 可以在 OpenXLA/xla 代码库中提交问题和功能请求。
- 如有关于 PJRT 的问题,请在 OpenXLA Discord 上提问。
PJRT C API 是我们希望添加到机器学习生态系统中的统一设备 API。 我们的长期愿景是:(1) 框架(TF、JAX 等)将调用 PJRT,PJRT 具有对框架不透明的设备专用实现;(2) 每台设备都专注于将 PJRT API 实现为 PJRT 插件,这些插件对框架不透明。
如未另行说明,那么本页面中的内容已根据知识共享署名 4.0 许可获得了许可,并且代码示例已根据 Apache 2.0 许可获得了许可。有关详情,请参阅 Google 开发者网站政策。Java 是 Oracle 和/或其关联公司的注册商标。
最后更新时间 (UTC):2024-11-13。