PJRT - Uniform Device API

PJRT C API는 ML 생태계에 추가하려는 통합 Device API입니다. 장기적인 비전은 (1) 프레임워크 (TF, JAX 등)가 프레임워크에 불투명한 기기별 구현이 있는 PJRT를 호출하고 (2) 각 기기가 프레임워크에 불투명할 수 있는 PJRT API를 PJRT 플러그인으로 구현하는 데 중점을 두는 것입니다.

통신 채널

리소스