PJRT - Uniform Device API

PJRT C API همان Device API است که می خواهیم به اکوسیستم ML اضافه کنیم. چشم انداز بلندمدت این است که: (1) فریم ورک ها (TF، JAX، و غیره) PJRT را فراخوانی می کنند، که پیاده سازی های خاص دستگاه دارد که نسبت به فریم ورک ها غیرشفاف هستند. (2) هر دستگاه بر پیاده سازی API های PJRT به عنوان پلاگین های PJRT تمرکز دارد که می تواند برای چارچوب ها غیر شفاف باشد.

کانال های ارتباطی

  • مشکلات و درخواست های ویژگی را می توان در مخزن OpenXLA/xla ثبت کرد.
  • سوالات مربوط به PJRT را می توان در OpenXLA Discord پرسید.

منابع