AWS ने अपने Trainium AI चिप्स के लिए PyTorch के नेटिव बैकएंड TorchNeuron का सार्वजनिक बीटा घोषित किया। यह ओपन-सोर्स लाइब्रेरी डेवलपर को सिर्फ़ .to('neuron') जोड़कर Trainium पर PyTorch कोड चलाने देती है। यह eager execution और torch.compile जैसी PyTorch की मानक सुविधाओं के साथ काम करती है। TorchNeuron, GitHub रिपॉज़िटरी के ज़रिए EC2 Trainium2 और Trainium3 इंस्टेंस के लिए उपलब्ध है।
यह क्यों मायने रखता है
अब शोधकर्ता कोड दोबारा लिखे बिना AWS के हाई-परफॉर्मेंस Trainium चिप्स पर PyTorch मॉडल ट्रेन कर सकते हैं, जिससे लागत और मेहनत कम होगी।
