فتحت جوجل المصدر لمشروع TorchTPU، وهو واجهة خلفية لـ PyTorch تتيح للمستخدمين تشغيل نماذج وعمليات PyTorch على وحدات معالجة الموتر (TPU) التابعة لجوجل. يتضمن المشروع نوى ATen مخصصة لمجموعة من عمليات PyTorch، وذاكرة تخزين مؤقت للترجمة لتحسين تنفيذ حسابات XLA، وأمثلة للتدريب الموزع والاستدلال على النماذج.
تشمل الميزات الرئيسية دعم التدريب الموزع عبر التوازي في البيانات والتوازي في الموترات، والتكامل مع torch.compile() لتنفيذ نمط الرسم البياني، وأدوات للتصحيح والقياس المعياري. وتغطي النماذج التجريبية Llama وQwen وResNet وDLRM وminGPT وUNet.
يتوفر دليل للمستخدم ودفاتر Marimo تفاعلية، وهي شبيهة بدفاتر Jupyter. يجب تشغيل الدفاتر من جهاز افتراضي لوحدة TPU، ويُوصى باستخدام مثيل v6e GCE كخيار أسهل. ويُدعم حالياً الإصدار Python 3.11 والإصدارات الأحدث، بما يتوافق مع مصفوفة توافق إصدارات PyTorch.
يمكن تثبيت أحدث إصدار عام، وهو 0.2.0، من PyPI باستخدام pip، مع استخدام إصدار PyTorch المخصص للمعالج المركزي من فهرس إضافي. ويمكن للمطورين أيضاً بناء ملفات wheel من المصدر باستخدام Bazel، مع تفعيل البناءات المحسّنة افتراضياً وخيار لتعطيل Remote Build Execution لمن لا تتوفر لديهم بيانات الاعتماد.
المصدر: Hacker News · لخّصه HeadlinesBriefing