AWS Inferentia2 vs AWS Trainium2

The Trainium2 delivers 6.8x the FP8 throughput of the Inferentia2 (1,299 vs 190 TFLOPS dense). The Trainium2 also carries 64 GB more memory (96 GB vs 32 GB).

Inferentia2: NeuronCore-v2, 2022 Trainium2: NeuronCore-v3, 2024

FP8 dense lead

6.8x

Trainium2: 1,299 vs 190 TFLOPS

Memory

32 vs 96 GB

64 GB more for the Trainium2

Bandwidth

820 GB/s vs 2.9 TB/s

Trainium2 moves data faster

Download comparison sheet Printable version →

AWS Inferentia2

Full specs →

What is the AWS Inferentia2?

The Inferentia2 is an ASIC from AWS on the NeuronCore-v2 architecture, an accelerator built for AI workloads. Inferentia2 full specifications →

Can I rent the Inferentia2 in the cloud?

Flopper is not tracking any live rental listings for the Inferentia2 yet. The provider directory lists the clouds that rent comparable accelerators on demand. GPU cloud providers →

How does the Inferentia2 compare with other accelerators?

This page sets the Inferentia2 against the Trainium2. The Inferentia2 is the older generation of the two. Both are AWS parts. Flopper also compares it head to head with the L4 24GB, across every published precision, memory and power. Inferentia2 vs L4 24GB →

Where do the Inferentia2 figures come from?

Every Inferentia2 figure on this page traces to AWS's own datasheet or specification table. How Flopper verifies specifications →

AWS Trainium2

Leads on FP8 Full specs →

What is the AWS Trainium2?

The Trainium2 is an ASIC from AWS on the NeuronCore-v3 architecture, an accelerator built for AI workloads. Trainium2 full specifications →

Can I rent the Trainium2 in the cloud?

Flopper is not tracking any live rental listings for the Trainium2 yet. The provider directory lists the clouds that rent comparable accelerators on demand. GPU cloud providers →

How does the Trainium2 compare with other accelerators?

This page sets the Trainium2 against the Inferentia2. The Trainium2 is the newer generation of the two. Both are AWS parts. Flopper also compares it head to head with the H100 SXM5 80GB, across every published precision, memory and power. Trainium2 vs H100 SXM5 80GB →

Where do the Trainium2 figures come from?

Every Trainium2 figure on this page traces to AWS's own datasheet or specification table. How Flopper verifies specifications →

Specifications

Inferentia2

Trainium2

Architecture

NeuronCore-v2NeuronCore-v3

Launch Year

20222024

Form Factor

——

Memory

32 GB96 GB

Memory Bandwidth

820 GB/s2.9 TB/s

TDP

——

Process Node

——

Spec

Inferentia2

Trainium2

Architecture NeuronCore-v2NeuronCore-v3

Launch Year 20222024

Form Factor ——

Memory 32 GB96 GB

Memory Bandwidth 820 GB/s2.9 TB/s

TDP ——

Process Node ——

Performance (TFLOPS)

Inferentia2

Trainium2

48 TFLOPS 181 TFLOPS

190 TFLOPS 667 TFLOPS

2,563 TFLOPS sparse

190 TFLOPS 667 TFLOPS

2,563 TFLOPS sparse

190 TFLOPS 667 TFLOPS

2,563 TFLOPS sparse

190 TFLOPS 1,299 TFLOPS

2,563 TFLOPS sparse

FP6

380 TOPS

Precision

Inferentia2

Trainium2

48 TFLOPS 181 TFLOPS

190 TFLOPS 667 TFLOPS

2,563 TFLOPS sparse

190 TFLOPS 667 TFLOPS

2,563 TFLOPS sparse

190 TFLOPS 667 TFLOPS

2,563 TFLOPS sparse

190 TFLOPS 1,299 TFLOPS

2,563 TFLOPS sparse

FP6

380 TOPS

Figures are dense. Where a vendor publishes a figure it sits under the dense one and never decides a winner. Hover any precision for its definition, or read the GPU glossary.

FLOPS by Precision

What actually differs

The Trainium2 is the newer part: NeuronCore-v3 (2024) against the Inferentia2's NeuronCore-v2 (2022).

The Trainium2 carries 64 GB more memory (96 GB against 32 GB), so larger models fit on one GPU before you shard. Bandwidth is 820 GB/s against 2.9 TB/s, which sets the ceiling on decode speed.

Weight the FP8 and FP16 rows for LLM work and the FP64 row for HPC.

New to the terms? The glossary explains FP32, TF32, BF16, FP16, FP8, INT8, Sparsity and TFLOPS.

Frequently asked questions

What is better, the Inferentia2 or the Trainium2?

The Trainium2, on throughput. At FP8 precision the Trainium2 reaches 1,299 TFLOPS dense against 190 TFLOPS for the Inferentia2. The performance table on this page lists every published precision for both GPUs.

Which should you choose, the Inferentia2 or the Trainium2?

Choose the Trainium2 for training and high-throughput inference: it has 6.8x the FP8 throughput of the Inferentia2. It also carries 64 GB more memory, so larger models fit on one GPU before you shard.

Which has more memory, the Inferentia2 or the Trainium2?

The Trainium2 carries 96 GB of memory versus 32 GB for the Inferentia2. Memory bandwidth is 820 GB/s for the Inferentia2 and 2.9 TB/s for the Trainium2.

Get Comparison Updates

New GPUs added weekly. Be the first to see how they compare.

Open in Full Compare Tool → Inferentia2 full specs Trainium2 full specs Estimate cluster cost in the calculator →

AWS Inferentia2 vs AWS Trainium2

AWS Inferentia2

What is the AWS Inferentia2?

Can I rent the Inferentia2 in the cloud?

How does the Inferentia2 compare with other accelerators?

Where do the Inferentia2 figures come from?

AWS Trainium2

What is the AWS Trainium2?

Can I rent the Trainium2 in the cloud?

How does the Trainium2 compare with other accelerators?

Where do the Trainium2 figures come from?

Specifications

Performance (TFLOPS)

FLOPS by Precision

What actually differs

Frequently asked questions

What is better, the Inferentia2 or the Trainium2?

Which should you choose, the Inferentia2 or the Trainium2?

Which has more memory, the Inferentia2 or the Trainium2?

Related comparisons

Get Comparison Updates

Stay Updated on GPU Specs