ข้ามไปยังเนื้อหา

แท็ก

#NKI

1 บทความ

AWS เปิดซอร์ส TorchNeuron: แค่ .to('neuron') โค้ด PyTorch ก็รันบน Trainium ได้ — public beta (Apache-2.0) แต่ torch.compile รับแค่ shape คงที่, op ที่ยังไม่รองรับแอบไปรันบน CPU และมี wheel ให้แค่ Ubuntu 26.04
เครื่องมือนักพัฒนา

AWS เปิดซอร์ส TorchNeuron: แค่ .to('neuron') โค้ด PyTorch ก็รันบน Trainium ได้ — public beta (Apache-2.0) แต่ torch.compile รับแค่ shape คงที่, op ที่ยังไม่รองรับแอบไปรันบน CPU และมี wheel ให้แค่ Ubuntu 26.04

AWS เปิดตัว TorchNeuron (19 ต.ค. 2026) และปล่อยเป็นโอเพนซอร์ส Apache-2.0 แบบ public beta ในวันที่ 20 ต.ค. เป็น backend ของ PyTorch ที่เพิ่ม device ชื่อ neuron ผ่าน PrivateUse1 ให้โค้ดเดิมรันบน Trainium2/3 ได้ทั้ง eager, torch.compile และ distributed แต่ beta นี้ยังมีข้อจำกัดชัดเจน คือ torch.compile ไม่รองรับ dynamic shape, op ที่ยังไม่รองรับจะ fallback ไปรันบน CPU, float64 ถูกลดเป็น float32 และ wheel สำเร็จรูปมีให้เฉพาะ Ubuntu 26.04

· อ่าน 5 นาที