RSS Блог разработчиков Google
Подписаться
TorchTPU: Нативное выполнение PyTorch на TPU в Google масштабе
TorchTPU — это новый инженерный стек, разработанный для обеспечения нативного, высокопроизводительного опыта запуска рабочих нагрузок PyTorch на инфраструктуре TPU Google с минимальными изменениями в коде. Он имеет подход "Eager First" с несколькими режимами выполнения и использует компилятор XLA для оптимизации распределенного обучения в огромных кластерах. К 2026 году проект нацелен на дальнейшее снижение накладных расходов на компиляцию и расширение поддержки динамических форм и пользовательских ядер для обеспечения бесшовной масштабируемости для следующего поколения ИИ.
