Chip comparison
AWS Trainium 2 vs Google TPU v4
Side-by-side specs
Straight from each vendor's datasheet. Winner in bold: higher throughput, memory, or transistor count; lower power draw or price. Sparse (2:4) throughput roughly doubles the FP8/INT8 numbers shown.
| Field | AWS Trainium 2 | Google TPU v4 |
|---|---|---|
| Process node | TSMC N5 | TSMC N7 |
| TDP | 500 W | 192 W |
| Memory | 96 GB HBM3 | 32 GB HBM2 |
| Memory bandwidth | 2,900 GB/s | 1,200 GB/s |
| BF16 (dense) | 667 TFLOPS | 275 TFLOPS |
| FP8 (dense) | 1,300 TFLOPS | — |
| INT8 (dense) | 1,300 TOPS | 275 TOPS |
| Form factor | AWS Neuron 2 (Trn2 UltraServer with 64 chips per unit) | OAM (per-chip in v4 pod) |
| Announced | 2024-12-03 | 2021-05-18 |
| Released | 2025-01-01 | 2022-05-01 |
Robots running AWS Trainium 2
No robots publicly running it yet.
Robots running Google TPU v4
No robots publicly running it yet.
Data centers with AWS Trainium 2
Data centers with Google TPU v4
Common questions
AWS Trainium 2 vs Google TPU v4: which is faster for training?
AWS Trainium 2 has 4.73x the dense FP16/BF16 throughput of the other (AWS Trainium 2: 1,300 TFLOPS; Google TPU v4: 275 TFLOPS). Real-world training speed also depends on memory bandwidth, interconnect, and software stack: see the spec diff for those inputs.
AWS Trainium 2 vs Google TPU v4: which has more memory?
AWS Trainium 2 carries more HBM (AWS Trainium 2: 96 GB HBM3; Google TPU v4: 32 GB HBM2). Memory capacity sets the largest model that fits on one chip without partitioning; memory bandwidth sets inference throughput.
AWS Trainium 2 vs Google TPU v4: which is more power-efficient?
Dense FP16/BF16 TFLOPS per watt: AWS Trainium 2 2.60 (1,300 ÷ 500 W); Google TPU v4 1.43 (275 ÷ 192 W). AWS Trainium 2 wins at the die level; system-level efficiency also depends on cooling and interconnect.
AWS Trainium 2 vs Google TPU v4: which is more widely used?
AWS Trainium 2: 0 robots and 3 data centers publicly running it. Google TPU v4: 0 robots and 2 data centers. Empty here means no operator has said publicly, not zero adoption in the market.
See also: every chip comparison · AWS Trainium 2 full page · Google TPU v4 full page.