हुआवेई एस्केंड की कंप्यूटिंग सिस्टम टीम ने 2.8 ट्रिलियन पैरामीटर वाले Kimi K3.1 Pro मॉडल के इन्फेरेंस के लिए अनुकूलन पूरा करने की घोषणा की। IT Home और ifeng Tech द्वारा रिपोर्ट किए जाने के अनुसार, Moonshot AI ने 4 नवंबर को इस मॉडल के पूर्ण वज़न जारी किए थे, जैसा कि हुआवेई कंप्यूटिंग के आधिकारिक खाते पर बताया गया था।
vLLM Ascend और SGLang का उपयोग करते हुए, डेवलपर्स अब इसे Atlas 800 A3 सर्वर, Atlas 900 A3 SuperPoD क्लस्टर और इसके लॉन्च के बाद से Ascend 950 SuperPoD श्रृंखला पर तैनात कर सकते हैं।
Kimi K3.1 Pro Moonshot AI की K3.1 श्रृंखला का पहला मॉडल है। यह एक मिश्रण-ऑफ-एक्सपर्ट्स प्रकार का मॉडल है जिसमें नेटिव विज़ुअल प्रोसेसिंग और लगभग दस लाख टोकन की संदर्भ विंडो है, जो इन्फेरेंस के प्रत्येक चरण में लगभग 104 बिलियन पैरामीटर सक्रिय करता है। मॉडल K3 पर आधारित हाइब्रिड लीनियर अटेंशन को बनाए रखता है, जो Kimi Delta Attention (KDA) लेयर्स पर आधारित है, और लंबी संदर्भ मॉडलिंग की लागत को कम करने के लिए कुछ फुल अटेंशन लेयर्स के स्थान पर DSA स्पार्स अटेंशन लेयर्स जोड़ता है।
अनुकूलन का अधिकांश काम विशेष रूप से इस आर्किटेक्चर से संबंधित था। हुआवेई ने कहा कि उसकी इन्फेरेंस टीम ने पिछली Kimi K3 अनुकूलन पर भरोसा किया, नए मॉडल का पूर्व विश्लेषण किया और समानांतर में मॉडल एकीकरण, ऑपरेटर विकास और सेवा सत्यापन पर काम किया। KDA और MoE के पहले से मौजूद समर्थन के कारण, टीम ने मौजूदा इन्फेरेंस फ्रेमवर्क में सेवा कार्यों को बनाए रखते हुए एक नया स्पार्स अटेंशन पाथ और मल्टीमॉडल इन्फेरेंस जोड़ा।
हुआवेई से मिली जानकारी के अनुसार, अनुकूलन में ऐतिहासिक जानकारी चयन, स्पार्स अटेंशन गणना, प्रीप्रोसेसिंग और अवशेष गणना के लिए मुख्य ऑपरेटरों को शामिल किया गया था, जिसे लंबे संदर्भ इन्फेरेंस और उत्पादन रखरखाव का समर्थन करने के लिए वितरित कैशिंग और फ्रेमवर्क स्तर के कार्यों के साथ जोड़ा गया था। कंपनी का लक्ष्य यह निर्धारित करना था कि उपयोगकर्ता और डेवलपर मॉडल के रिलीज के पहले दिन से ही एप्लिकेशन परीक्षण और परिनियोजन शुरू कर सकें।
vLLM Ascend दस्तावेज़ीकरण इंगित करता है कि Kimi K3.1 Pro को vLLM Ascend 0.30.0 संस्करण में प्रायोगिक मोड में समर्थित किया जाता है। इसकी संदर्भ संरचना आठ नोड्स पर प्रीफाइल और डिकोडिंग विभाजन का उपयोग करती है, जिसमें चार प्रीफाइल नोड्स और चार डिकोडिंग नोड्स शामिल हैं, जिसमें एक मिलियन टोकन की लंबी संदर्भ कॉन्फ़िगरेशन भी शामिल है। Atlas 800I A3 सर्वर W4A8 क्वांटाइज्ड चेकपॉइंट के साथ काम करते हैं, जबकि 950DT उत्पाद W4A8 MXFP संस्करण का उपयोग करते हैं; दिशानिर्देश बताते हैं कि Atlas A2 और 950PR उत्पाद अभी तक समर्थित नहीं हैं।
मॉडल का आकार हार्डवेयर समर्थन को एक व्यावहारिक प्रश्न बनाता है। IT Home बताता है कि BF16 प्रारूप में वज़न लगभग 5.6 TB लेते हैं और INT4 क्वांटाइजेशन पर अभी भी लगभग 1.4 TB की आवश्यकता होती है, जो एक उपभोक्ता जीपीयू का उपयोग करने को असंभव बना देता है। हुआवेई ने अपनी A3 और 950 SuperPoD प्रणालियों को इस ट्रिलियन-पैरामीटर MoE इन्फेरेंस वर्ग के लिए तैनात किया है, और Kimi K3.1 Pro का रिलीज उस दिन समर्थित अंतिम ओपन मॉडल था।



