Oossa

TorchNeuron يضيف دعمًا أصليًا لـPyTorch على AWS Trainium

أطلقت PyTorch مكتبة TorchNeuron مفتوحة المصدر، تتيح تشغيل النماذج على شرائح Trainium من دون تغيير الشيفرة.

خبر موجزبقلم نشرته Oossa: آخر تحديث: 1 دقائق قراءة

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

لماذا يهم

يمكن للباحثين استخدام شيفرة PyTorch نفسها للاستفادة من سرعة Trainium من دون إعادة كتابة سير العمل.

هل كان هذا المقال مفيدًا؟
مشاركة

اقرأ أيضًا

Oossa · النشرة البريدية

أسبوع الذكاء الاصطناعي، مشروحًا

كل يوم اثنين: الأخبار التي تستحق المعرفة، بلغة بسيطة. مجانًا وبلا رسائل مزعجة.