Existing Likely

Paper on Switch transformer

Google • January 11, 2021
Total GPUs
1.0K
H100 Equiv
63
Power
0.5 MW
FP16
126 PF

Specifications

Primary GPUGoogle TPU v3
GPU Count1.0K
SectorPrivate

Additional Information

The TPU count is from the expert parallelism (1 expert per core) with 2048 experts and 2 tensor cores per TPU v3.