Mojo داخل رسم MAX البياني

يُمثَّل النموذج داخل MAX على شكل رسم بياني: شبكة من العمليات تنتقل فيها البيانات من عقدة إلى العقدة التالية. 🔗

كل عقدة في الرسم البياني هي op، أي عملية واحدة مثل ضرب المصفوفات أو دالة تنشيط تُطبَّق على مدخلاتها.

يأتي MAX مزودًا بالعديد من العمليات المضمنة التي تغطي الطبقات الشائعة، لذلك يعمل معظم الرسم البياني للنموذج المعتاد من دون شيفرة إضافية.

قد يحتاج نموذجك أحيانًا إلى عملية لا يتضمنها MAX. عندها تكتب op مخصصة لسد هذه الفجوة.

تكتب العمليات المخصصة باستخدام Mojo، فتحصل على سرعة بمستوى الأنظمة للحساب الدقيق الذي يتطلبه نموذجك.

العملية المخصصة هي في الأساس دالة في Mojo تأخذ موترات الإدخال وتنتج موتر إخراج للرسم البياني.

بما أن العملية مكتوبة بـMojo، يمكنك استخدام SIMD وحيل أخرى لجعل تلك العقدة تعمل بأقصى سرعة يسمح بها العتاد.

بعد تعريف العملية المخصصة، تُجرى عملية تسجيل لها حتى يتمكن MAX من وضعها في الرسم البياني مثل أي عملية مضمنة.

يواصل MAX تحسين الرسم البياني الكامل، بما في ذلك عقدتك المخصصة، من خلال دمج العمل وجدولته لتحقيق أفضل إنتاجية إجمالية.

يتيح لك هذا النمط إبقاء النموذج قياسيًا في معظمه، مع إدراج عملية Mojo مضبوطة يدويًا في الموضع الذي تحتاج فيه إلى السرعة.

استخدم عملية مخصصة عندما تكون عقدة مفقودة أو بطيئة هي عنق الزجاجة لديك، لا في الأجزاء التي تعالجها العمليات المضمنة جيدًا. ✨

استرجع كيفية اتصال Mojo بالرسم البياني للحساب في MAX.

رأيت أن نموذج MAX هو رسم بياني من العمليات، وأن Mojo تتيح لك كتابة عمليات مخصصة سريعة تُدرج مباشرة فيه. 🎯