- यह एक पैकेज है जो Go भाषा में machine learning (ML) मॉडल लागू करने देता है, और इसे इस तरह डिज़ाइन किया गया है कि Python पर निर्भर हुए बिना भी ML कार्य किए जा सकें
- यह TensorFlow(TF), JAX, PyTorch जैसे Python-केंद्रित ML framework के विकल्प के रूप में काम करता है, और वही hardware efficiency Go-आधारित तरीके से उपयोग करता है
- यह XLA और PJRT जैसी low-level compiler और runtime तकनीकों का उपयोग करके optimized performance प्रदान करता है
मुख्य विशेषताएँ
- Python के बिना ML मॉडल बनाना
- Python code के बिना केवल Go भाषा से ML मॉडल बनाना, training करना और inference चलाना संभव
- मौजूदा ML framework की expressive power के समान तरीके से computation graph बनाना, automatic differentiation और model training का समर्थन
- XLA और PJRT का उपयोग
- OpenXLA stack का उपयोग करके hardware-optimized operations को support करता है, और TensorFlow तथा JAX में इस्तेमाल होने वाले वही मूल building blocks का उपयोग करता है
- automatic differentiation, tensor data management, work partitioning जैसी advanced features शामिल
- CNN मॉडल implementation का उदाहरण
- CIFAR-10 dataset का उपयोग करके CNN(convolutional neural network) मॉडल को Go में लागू किया गया
- GoMLX से Python के बिना लागू किया गया मॉडल, TensorFlow+Keras से train किए गए मॉडल के समान प्रदर्शन हासिल करता है
- वास्तविक उदाहरण: Gemma2 मॉडल
- HuggingFace द्वारा उपलब्ध कराए गए Gemma2 LLM(Large Language Model) weights का उपयोग करके वास्तविक production-grade LLM inference चलाया गया
- Python के बिना भी Gemma2 जैसे advanced मॉडल चलाए जा सकते हैं
सारांश
- GoMLX, Go भाषा के आधार पर ML मॉडल implementation, training और inference को support करता है और Python dependency को हटाता है
- XLA और PJRT का उपयोग करके hardware-optimized operations और advanced ML features प्रदान करता है
- Python के बिना भी CIFAR-10 CNN मॉडल और Gemma2 LLM inference जैसे जटिल कार्य संभव हैं
- GoMLX अभी शुरुआती चरण में है, लेकिन यह Python-free ML implementation की संभावनाएँ खोलता है और आगे इसके विकास की उम्मीद है
अभी कोई टिप्पणी नहीं है.