أعلنت Hugging Face عن TorchTPU، وهي واجهة خلفية أصلية لـ PyTorch تتعامل مع وحدات TPU كأي جهاز آخر. ثبّتها بأمر pip واحد، ويمكنك تحميل نماذج Transformers أو Diffusers على وحدة TPU عبر ضبط device_map="tpu". تتضمن الحزمة نوى Pallas مُحسّنة لوحدات TPU، ودعمًا للتدريب الموزع، وتكاملًا مع torch.compile لزيادة السرعة. ولم تعد Optimum TPU مطلوبة لتشغيل نماذج Hugging Face على وحدات TPU.
لماذا يهم
بات بإمكان مطوري PyTorch استخدام وحدات TPU من Google Cloud من دون أطر عمل إضافية، ما يبسّط التدريب والاستدلال على نطاق واسع.
