पॉलीमेज लैब्स और टेन्सटॉरेंट ने AI कंपाइलर बनाने के लिए किया समझौता

भारतीय विज्ञान संस्थान (IISc) में विकसित एक डीप-टेक स्टार्टअप, पॉलीमेज लैब्स ने टेन्सटॉरेंट के एआई एक्सेलेरेटर के लिए एक एआई कंपाइलर विकसित करने के लिए अमेरिका स्थित टेन्सटोरेंट इंक के साथ एक रणनीतिक सहयोग किया है।

इस साझेदारी के तहत पॉलीमेज लैब्स की कंपाइलर विशेषज्ञता और टेन्सटॉरेंट के एआई कंप्यूटिंग प्लेटफॉर्म को एक साथ लाया गया है, जिसका उद्देश्य उन्नत एआई हार्डवेयर के लिए सॉफ्टवेयर समर्थन को बेहतर बनाना है। दोनों कंपनियों ने कहा कि यह सहयोग विशेषीकृत एआई चिप्स को प्रोग्राम करने और तैनात करने में आने वाली लंबे समय से चली आ रही चुनौतियों का समाधान करने पर केंद्रित है।

पॉलीमेज लैब्स के अनुसार, कंपाइलर सॉफ्टवेयर डेवलपर्स द्वारा लिखे गए उच्च-स्तरीय निर्देशों को अनुकूलित कोड में बदलने में महत्वपूर्ण भूमिका निभाता है, जो जटिल हार्डवेयर पर कुशलतापूर्वक चल सकता है। कंपनी ने कहा कि इस तरह का सॉफ्टवेयर अगली पीढ़ी के एआई एक्सेलरेटर के उपयोग को सरल बनाने के लिए आवश्यक है, जिनमें अक्सर परिष्कृत प्रोग्रामिंग दृष्टिकोण की आवश्यकता होती है।

साथ मिलकर काम करने के कुछ ही महीनों के भीतर, पॉलीमेज लैब्स के पॉलीब्लॉक्स कंपाइलर फ्रेमवर्क को टेनस्टोरेंट के हार्डवेयर प्लेटफॉर्म के साथ एकीकृत कर दिया गया है। कंपनी ने कहा कि यह एकीकरण मशीन लर्निंग और एआई अनुसंधान में व्यापक रूप से उपयोग किए जाने वाले दो फ्रेमवर्क, PyTorch और JAX कोड से सीधे, बिना किसी बदलाव के, उच्च-स्तरीय कोड से ही दमदार प्रदर्शन प्रदान करता है।

"पॉलीमेज और टेनस्टोरेंट टीमों के बीच तालमेल असाधारण रहा है" पॉलीमेज लैब्स के चीफ टेक्नोलॉजी ऑफिसर और IISC में कंप्यूटर विज्ञान के प्रोफेसर उदय बोंधुगुला ने कहा "पॉलीब्लॉक्स और टेनस्टोरेंट के tt-mlir दोनों में आधुनिक MLIR-आधारित कंपाइलर इन्फ्रास्ट्रक्चर का उपयोग करके, हम कुछ ही महीनों में PyTorch और JAX के लिए एक संपूर्ण, एंड-टू-एंड कंपाइलर विकसित करने में सक्षम हुए।"

PyTorch एक ओपन-सोर्स डीप लर्निंग फ्रेमवर्क है जिसे मूल रूप से मेटा प्लेटफॉर्म्स द्वारा विकसित किया गया था और लिनक्स फाउंडेशन द्वारा समर्थित है, जबकि JAX एक पायथन-आधारित लाइब्रेरी है जिसे उच्च-प्रदर्शन संख्यात्मक कंप्यूटिंग और बड़े पैमाने पर मशीन लर्निंग के लिए डिज़ाइन किया गया है।

इस सहयोग का उद्देश्य एआई हार्डवेयर उद्योग में एक प्रमुख बाधा को दूर करना भी है। एक परिपक्व और निर्बाध सॉफ्टवेयर इकोसिस्टम की कमी, जिसने कई एक्सेलेरेटर प्लेटफार्मों को अपनाने की गति को धीमा कर दिया है।