أعلنت PyTorch عن TorchNeuron، وهي واجهة خلفية أصلية تجعل AWS Trainium يظهر كجهاز عادي في PyTorch. وتتيح المكتبة الجديدة للمطورين نقل النموذج إلى Trainium ببساطة باستخدام .to('neuron')، والاستفادة من torch.compile لتحقيق سرعة مناسبة لبيئات الإنتاج. وهي تدعم التنفيذ الفوري، والوضع الفوري التكيّفي، والنوى المخصّصة عبر واجهة Neuron Kernel Interface. وستُنشر الشيفرة مفتوحة المصدر بموجب ترخيص Apache 2.0، على أن يبدأ الإصدار التجريبي العام في 20 أكتوبر.
لماذا يهم
يمكن للباحثين استخدام شيفرة PyTorch نفسها للاستفادة من سرعة Trainium من دون إعادة كتابة سير العمل.