PJRT – Uniform Device API

PJRT C API to jednolity interfejs Device API, który chcemy dodać do ekosystemu ML. W dłuższej perspektywie: (1) platformy (TF, JAX itp.) będą wywoływać PJRT, które mają implementacje specyficzne dla danego urządzenia, które są niejasne dla platformy; (2) każde urządzenie skupia się na implementowaniu interfejsów PJRT jako wtyczek PJRT, które mogą być przezroczyste dla platform.

Kanały komunikacji

Zasoby