PJRT — унифицированный API устройств

PJRT C API — это унифицированный API устройств, который мы хотим добавить в экосистему машинного обучения. В долгосрочной перспективе: (1) фреймворки (TF, JAX и т. д.) будут вызывать PJRT, который имеет реализации для конкретных устройств, непрозрачные для фреймворков; (2) каждое устройство ориентировано на реализацию API-интерфейсов PJRT в виде плагинов PJRT, которые могут быть непрозрачны для платформ.

Каналы связи

Ресурсы