PJRT - Uniform Device API

PJRT C API คือ Device API รูปแบบเดียวกันที่เราต้องการเพิ่มลงในระบบนิเวศ ML วิสัยทัศน์ระยะยาวคือ (1) เฟรมเวิร์ก (TF, JAX ฯลฯ) จะเรียกใช้ PJRT ซึ่งมีการใช้งานเฉพาะอุปกรณ์ที่เฟรมเวิร์กมองไม่เห็น (2) อุปกรณ์แต่ละเครื่องมุ่งเน้นที่การใช้ PJRT API เป็นปลั๊กอิน PJRT ซึ่งเฟรมเวิร์กมองไม่เห็น

ช่องทางการสื่อสาร

  • คุณสามารถรายงานปัญหาและส่งคำขอฟีเจอร์ใน OpenXLA/xla repo
  • คุณสามารถถามคําถามเกี่ยวกับ PJRT ได้ใน OpenXLA Discord

แหล่งข้อมูล