โมเดลภาษาขนาดใหญ่ AI
AXIOM
AXIOM เป็นเคอร์เนล Rust ที่สามารถบูตได้ ซึ่งปรับประสิทธิภาพการอนุมานของทรานส์ฟอร์เมอร์โดยการแทนที่นามธรรมของระบบปฏิบัติการทั่วไปด้วยพรีมิทีฟเฉพาะสำหรับการอนุมาน
AXIOM
AXIOM คืออะไร
AXIOM เป็นเคอร์เนลระบบปฏิบัติการวิจัยที่ออกแบบมาเพื่อรันงานอนุมานทรานส์ฟอร์เมอร์อย่างมีประสิทธิภาพบนฮาร์ดแวร์ที่มีหน่วยความจำจำกัด โดยใช้การจัดสรรหน่วยความจำแบบ tensor-native การจัดตารางรอบขอบเขตเลเยอร์ และการสตรีมน้ำหนักแบบ double-buffered
AXIOM เทียบกับเครื่องมือที่คล้ายคลึง
| โมเดลราคา | ฟรี | ฟรี | ฟรี | ฟรี |
| เครดิตฟรี | ||||
| ฟีเจอร์หลัก |
|
|
|
|
| ข้อดี |
|
|
|
|
| ข้อจำกัด |
|
|
|
|
| เหมาะสำหรับ |
|
|
|
|
วิธีใช้ AXIOM
- 1ตั้งค่า Rust nightly toolchain และติดตั้ง bootimage
- 2บรรจุน้ำหนักโมเดลโดยใช้ Python script ที่ให้มา (pack_weights.py)
- 3สร้างเคอร์เนลด้วย cargo +nightly run --release
- 4บูตเคอร์เนลใน QEMU หรือบน bare metal; มันจะทำการอนุมานและแสดงผล telemetry
ฟีเจอร์หลักของ AXIOM
- การจัดสรรหน่วยความจำแบบ tensor-native พร้อมพูลที่จองล่วงหน้า
- การจัดตารางรอบขอบเขตเลเยอร์เพื่อป้องกันการแทรกกลางเลเยอร์
- การสตรีมน้ำหนักแบบ double-buffered เพื่อให้ I/O และการคำนวณทับซ้อนกัน
- ตัวจัดตาราง LayerLock สำหรับการดำเนินการที่ reside ในแคช
- การอนุมานแบบ quantized (Q4) สำหรับ SmolLM2-135M และ TinyLlama-1.1B
เคสใช้งานของ AXIOM
- การรันโมเดลภาษาขนาดใหญ่บนระบบที่มีหน่วยความจำจำกัด
- การปรับปรุงเวลาแฝงของการอนุมานสำหรับโมเดลทรานส์ฟอร์เมอร์
- การวิจัยเกี่ยวกับการเพิ่มประสิทธิภาพระดับระบบปฏิบัติการสำหรับงาน AI
- การประเมินผลกระทบของการจัดตารางระดับเคอร์เนลต่อปริมาณงานอนุมาน
ราคาและเครดิตฟรีของ AXIOM
AXIOM ใช้โมเดลราคาแบบ ฟรี
ข้อดีและข้อจำกัดของ AXIOM
ข้อดี
- ลดค่าใช้จ่ายในการสตรีมจากวินาทีเหลือไมโครวินาทีต่อเลเยอร์
- ปรับแต่งรูปแบบหน่วยความจำสำหรับรูปแบบการเข้าถึงอนุมานที่คาดเดาได้
- โอเพนซอร์สและขยายได้สำหรับการวิจัย
- แสดงให้เห็นถึงการปรับปรุงปริมาณงานอย่างมีนัยสำคัญ (14.8x TPS ในการวัดประสิทธิภาพ)
ข้อจำกัด
- ปัจจุบันจำกัดเฉพาะโมเดลที่เฉพาะเจาะจง (SmolLM2-135M, TinyLlama-1.1B Q4)
- ต้องใช้ NVMe แบบ bare metal สำหรับการประเมินคลาส 7B ที่ใช้หน่วยความจำต่ำตามที่ตั้งใจ
- เคอร์เนลการคำนวณ (FFN projection) ยังคงเป็นคอขวด
- ไม่ใช่ระบบปฏิบัติการทั่วไป; ขาด userspace, เครือข่าย, ระบบไฟล์
AXIOM เหมาะกับงานแบบไหน?
- นักวิจัยด้านระบบ AI และระบบปฏิบัติการ
- นักพัฒนาที่เพิ่มประสิทธิภาพการอนุมานบนฮาร์ดแวร์ที่มีข้อจำกัด
- วิศวกรที่สนใจวิศวกรรมประสิทธิภาพระดับเคอร์เนลสำหรับ AI