design proposal: PJRT C API optional and experimental features

146 views
Skip to first unread message

Skye Wanderman-Milne

unread,
Jan 18, 2024, 4:19:20 PM1/18/24
to pjrt-a...@googlegroups.com, openxla...@openxla.org, Jieying Luo, Will Cromar
Hi PJRT plugin authors + friends,

We'd like to introduce a structured way to handle optional and experimental features to the PJRT C API. This should hopefully make everyone's lives easier by clearly delineating what's most important to implement vs. leave unimplemented, and make sure there's graceful handling of unimplemented optional features. Brief design proposal here: https://docs.google.com/document/d/1PsSIkCbHujPfiZphyQ3U5wB9VGBLKBjxGrTtmziJwEw/edit

We're planning to implement handling of optional features in JAX first, but the mechanism is generic across frameworks.

We'd love any feedback you have on the proposal! Feel free to reply here or comment on the doc. 

Thanks,
Skye
Reply all
Reply to author
Forward
0 new messages