PJRT – Uniform Device API

Die PJRT C API ist die einheitliche Geräte-API, die wir dem ML-System hinzufügen möchten. Langfristig soll es so aussehen: (1) Frameworks (TF, JAX usw.) rufen PJRT auf, das gerätespezifische Implementierungen hat, die für die Frameworks nicht transparent sind. (2) Auf jedem Gerät werden PJRT APIs als PJRT-Plug-ins implementiert, die für die Frameworks nicht transparent sind.

Kommunikationskanäle

Ressourcen