Skip to main content

Google TPU overview

· 2 min read
Boris V.Kuznetsov
Neurodyne Founder

Обзор технологии TPU и линейки Google TPU v1-v7

Введение

V1

Так выглядит PCIe плата для V1

Boardv1

Линейка v1-v7

All

v2 на плате по 4шт в 0.5U сервере

V2

v2 vs v3

V2-vs-V3

v3

V3

v4

V4

А так выглядит силикон. Слева - 3D сборка TPU v4. Сам чип - в центре. По бокам 4 блока - это HBM2 RAM, которая подключается к TPU через широкие 1024-битные шины на общей подложке-интерпозере.

V4

Сетевая топология

Tori

Pod и деплой

Training pod и разбиение на слайсы

Slices

V5 Training Pod для инференса

Slices

Типовой деплой TPU где-то в Google DC

Slices