TorchTPU: нативный запуск PyTorch на TPU

Описание: TorchTPU — это совместный проект Google и Meta с открытым исходным кодом, представляющий собой нативный, основанный на принципах PyTorch бэкенд для TPU от Google. Он обеспечивает бесперебойное выполнение моделей «из коробки» и миграцию на TPU, одновременно предоставляя конкурентоспособную производительность при выводе и обучении за счет оптимизации выполнения, интеграции torch.compile и нативного аппаратного ускорения.

Клаудио Базиле из Google и Колин Тейлор из Meta рассказывают, как запускать нативные рабочие нагрузки PyTorch на TPU от Google с помощью TorchTPU.