PJRT : API Uniform Device

L'API C PJRT est l'API Device uniforme que nous souhaitons ajouter à l'écosystème ML. À long terme, les frameworks (TF, JAX, etc.) appelleront PJRT, qui dispose d'implémentations spécifiques à l'appareil qui sont opaques pour les frameworks. Chaque appareil se concentre sur l'implémentation des API PJRT en tant que plug-ins PJRT, qui peuvent être opaques pour les frameworks.

Canaux de communication

Ressources