DEPLOYDatabase

Chip comparison

AWS Trainium 2 vs Cerebras WSE-3 (Wafer-Scale Engine 3)

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.

FieldAWS Trainium 2Cerebras WSE-3 (Wafer-Scale Engine 3)
Process nodeTSMC N5TSMC N5
Transistors (B)4000
Die size46225 mm²
TDP500 W23000 W
Memory96 GB HBM344 GB on-die SRAM
Memory bandwidth2,900 GB/s21,000,000 GB/s
FP16 (dense)62,500 TFLOPS
BF16 (dense)667 TFLOPS
FP8 (dense)1,300 TFLOPS
INT8 (dense)1,300 TOPS
Form factorAWS Neuron 2 (Trn2 UltraServer with 64 chips per unit)wafer-scale (single-wafer system)
Announced2024-12-032024-03-13
Released2025-01-012024-04-01

Robots running AWS Trainium 2

No robots publicly running it yet.

Robots running Cerebras WSE-3 (Wafer-Scale Engine 3)

No robots publicly running it yet.

Data centers with Cerebras WSE-3 (Wafer-Scale Engine 3)

No data centers publicly running it yet.

Common questions

AWS Trainium 2 vs Cerebras WSE-3 (Wafer-Scale Engine 3): which is faster for training?

Cerebras WSE-3 (Wafer-Scale Engine 3) has 48.08x the dense FP16/BF16 throughput of the other (AWS Trainium 2: 1,300 TFLOPS; Cerebras WSE-3 (Wafer-Scale Engine 3): 62,500 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 Cerebras WSE-3 (Wafer-Scale Engine 3): which has more memory?

AWS Trainium 2 carries more HBM (AWS Trainium 2: 96 GB HBM3; Cerebras WSE-3 (Wafer-Scale Engine 3): 44 GB on-die SRAM). Memory capacity sets the largest model that fits on one chip without partitioning; memory bandwidth sets inference throughput.

AWS Trainium 2 vs Cerebras WSE-3 (Wafer-Scale Engine 3): which is more power-efficient?

Dense FP16/BF16 TFLOPS per watt: AWS Trainium 2 2.60 (1,300 ÷ 500 W); Cerebras WSE-3 (Wafer-Scale Engine 3) 2.72 (62,500 ÷ 23000 W). Cerebras WSE-3 (Wafer-Scale Engine 3) wins at the die level; system-level efficiency also depends on cooling and interconnect.

AWS Trainium 2 vs Cerebras WSE-3 (Wafer-Scale Engine 3): which is more widely used?

AWS Trainium 2: 0 robots and 3 data centers publicly running it. Cerebras WSE-3 (Wafer-Scale Engine 3): 0 robots and 0 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 · Cerebras WSE-3 (Wafer-Scale Engine 3) full page.