PJRT - Uniform Device API

A API PJRT C é a API Device uniforme que queremos adicionar ao ecossistema de ML. A visão de longo prazo é que: (1) os frameworks (TF, JAX etc.) vão chamar o PJRT, que tem implementações específicas do dispositivo que são opacas para os frameworks; (2) cada dispositivo se concentra na implementação de APIs PJRT como plug-ins PJRT, que podem ser opacas para os frameworks.

Canais de comunicação

Recursos