Google открыла исходный код TorchTPU для PyTorch на TPU

Google выложила на GitHub стек TorchTPU, добавляющий TPU (v5e, v5p, v6e и Ironwood) как нативный бэкенд PyTorch наряду с GPU и CPU. Проект создан совместно с командой PyTorch из Meta; вместе с ним в раннем превью открыт TPU Agent Suite для ИИ-агентов.

• TorchTPU ставится через pip и поддерживает Python 3.11+
• Работает с Hugging Face Transformers v5, vLLM, SGLang и TorchTitan
• TPU Agent Suite совместим с Claude Code и Antigravity, позже — Codex и Cursor
• Поддержаны TPU v5e, v5p, v6e и v7 Ironwood