PJRT: API de Uniform Device

La API de PJRT C es la API de dispositivos uniforme que queremos agregar al ecosistema de AA. La visión a largo plazo es que: (1) los frameworks (TF, JAX, etc.) llamarán a PJRT, que tiene implementaciones específicas para dispositivos que son opacas para los frameworks; (2) cada dispositivo se enfoca en implementar las APIs de PJRT como complementos de PJRT, que pueden ser opacos para los frameworks.

Canales de comunicación

Recursos