L'API PJRT C è l'API Device uniforme che vogliamo aggiungere all'ecosistema ML. La visione a lungo termine è che: (1) i framework (TF, JAX e così via) chiameranno PJRT, che ha implementazioni specifiche per dispositivo che sono opache per i framework; (2) ogni dispositivo si concentra sull'implementazione delle API PJRT come plug-in PJRT, che possono essere opachi per i framework.
Canali di comunicazione
- I problemi e le richieste di funzionalità possono essere archiviati nel repository OpenXLA/xla.
- Puoi porre domande relative a PJRT su Discord di OpenXLA.
Risorse
- Intestazione dell'API C PJRT
- Log delle modifiche dell'API C PJRT
- Guida all'integrazione di PJRT
- Documentazione sulla progettazione per PJRT
- Versionamento e compatibilità dell'ABI dell'API PJRT
- Documentazione sulla progettazione del meccanismo dei plug-in PJRT
- Implementazione del plug-in OpenXLA/IREE PJRT