أعلنت AWS إتاحة النسخة التجريبية العامة من TorchNeuron، وهي واجهة أصلية لـPyTorch مخصصة لشرائح Trainium. تتيح المكتبة مفتوحة المصدر للمطورين تشغيل شيفرة PyTorch على Trainium بمجرد إضافة .to('neuron'). وهي تدعم ميزات PyTorch القياسية، مثل التنفيذ الفوري وtorch.compile. تتوفر TorchNeuron لمثيلات EC2 Trainium2 وTrainium3 عبر مستودع على GitHub.
لماذا يهم
بات بإمكان الباحثين تدريب نماذج PyTorch على شرائح Trainium عالية الأداء من AWS من دون إعادة كتابة الشيفرة، ما يقلل التكلفة والجهد.