Google และ Meta เปิดซอร์ส TorchTPU แบ็คเอนด์ PyTorch
Google และ Meta ร่วมมือกันเปิดซอร์ส TorchTPU ซึ่งเป็นเนทีฟ PyTorch แบ็คเอนด์สำหรับ Cloud TPUs เพื่อเพิ่มประสิทธิภาพการทำงาน
ภาพประกอบจากคลังภาพสต็อก ไม่ใช่ภาพจากเหตุการณ์จริง
- Google และ Meta เปิดซอร์ส TorchTPU ซึ่งเป็นเนทีฟแบ็คเอนด์สำหรับ Cloud TPUs
- ระบบทำงานแบบ eager เป็นค่าเริ่มต้นและรองรับการคอมไพล์ผ่าน torch.compile กับ XLA
- พัฒนาขึ้นมาเพื่อใช้ทดแทน PyTorch/XLA สำหรับภาระงานใหม่ๆ โดยเฉพาะ
ในวงการปัญญาประดิษฐ์และการประมวลผลคลาวด์ มีความเคลื่อนไหวครั้งสำคัญเมื่อ Google และ Meta ได้ตัดสินใจร่วมกันเปิดซอร์ส TorchTPU ซึ่งเป็นเนทีฟ PyTorch แบ็คเอนด์ที่ออกแบบมาสำหรับ Cloud TPUs โดยเฉพาะ การปล่อยโค้ดชุดนี้ออกมาถือเป็นการเปลี่ยนแปลงครั้งใหญ่ในการพัฒนาและปรับแต่งโมเดล AI บนฮาร์ดแวร์เฉพาะทางของ Google ให้มีความคล่องตัวมากยิ่งขึ้น
สถาปัตยกรรมของ TorchTPU ถูกออกแบบมาให้ทำงานในลักษณะ eager mode เป็นค่าเริ่มต้น ซึ่งช่วยให้นักพัฒนาสามารถเขียนและทดสอบโค้ดได้อย่างเป็นธรรมชาติและรวดเร็ว นอกจากนี้ตัวระบบยังมีความสามารถในการคอมไพล์ผ่านเครื่องมือมาตรฐานอย่าง torch.compile และ XLA (Accelerated Linear Algebra) เพื่อรีดประสิทธิภาพสูงสุดออกมาจากการประมวลผลบน Cloud TPUs
การที่ยักษ์ใหญ่สองค่ายอย่าง Google และ Meta ร่วมมือกันผลักดันแบ็คเอนด์ตัวใหม่นี้ สะท้อนให้เห็นความพยายามในการลดความซับซ้อนของการใช้งานฮาร์ดแวร์เร่งความเร็ว AI นอกเหนือจากชิปค่ายหลัก การมีเนทีฟแบ็คเอนด์ที่รองรับ PyTorch โดยตรงจะช่วยให้นักวิจัยสามารถย้ายโมเดลมายัง Cloud TPUs ได้ง่ายขึ้นโดยไม่ต้องติดข้อจำกัดของเครื่องมือเก่าๆ
สำหรับการใช้งานจริง TorchTPU ถูกวางตำแหน่งให้เข้ามาแทนที่ระบบ PyTorch/XLA แบบเดิม สำหรับภาระงานและโปรเจกต์ใหม่ๆ ที่กำลังจะเริ่มต้นขึ้น ซึ่งจะช่วยลดภาระในการบำรุงรักษาและเพิ่มความเข้ากันได้กับฟีเจอร์ใหม่ๆ ของเฟรมเวิร์ก PyTorch ในอนาคต
ภาพประกอบจากคลังภาพสต็อก ไม่ใช่ภาพจากเหตุการณ์จริง
ที่มา: MarkTechPost
พบข้อมูลผิดพลาดในบทความนี้? แจ้งปัญหาบทความนี้
ความคิดเห็น
แสดงความคิดเห็น