PJRT - Uniform Device API

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

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

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

منابع