PJRT C API — это унифицированный API для устройств, который мы хотим добавить в экосистему машинного обучения. Долгосрочная перспектива такова: (1) фреймворки (TF, JAX и т. д.) будут вызывать PJRT, который имеет реализации, специфичные для устройств и непрозрачные для фреймворков; (2) каждое устройство будет ориентировано на реализацию API PJRT в виде плагинов PJRT, которые могут быть непрозрачными для фреймворков.
Каналы связи
- Сообщения об ошибках и предложения по улучшению можно отправлять в репозиторий OpenXLA/xla .
- Вопросы, касающиеся PJRT, можно задавать в Discord-канале OpenXLA .
Ресурсы
- Заголовочный файл API PJRT C
- Список изменений API PJRT C
- Руководство по интеграции PJRT
- проектная документация PJRT
- Версионирование и совместимость API ABI PJRT
- Документация по проектированию механизма плагина PJRT
- Презентация по плагину PJRT на осенней конференции OpenXLA 2024 ( запись )
- Реализация плагина IREE PJRT