AWS Trainium2
Overview
AWS Trainium2 is Amazon's third-generation AI training and inference accelerator, and the chip behind Trn2 instances and Trn2 UltraServers. Each chip carries eight NeuronCore-v3 delivering 1,299 FP8 TFLOPS, 667 TFLOPS across BF16, FP16 and TF32, and 181 FP32 TFLOPS, with 96 GiB of device memory at 2.9 TB/s and 224 MiB of software-managed on-chip SBUF scratchpad. Chip-to-chip traffic runs over NeuronLink-v3 at 1.28 TB/s per chip, and 16 dedicated CC-Cores orchestrate collective communication so that reductions do not consume compute cores. AWS publishes a sparse figure of 2,563 TFLOPS spanning FP8, FP16, BF16 and TF32; note that this is 1.97x the dense FP8 rate but 3.84x the dense BF16 rate, so it does not behave like conventional 2:4 structured sparsity and should not be read as a simple doubling. AWS publishes no TDP and does not name the process node or the memory generation for this chip.
Performance
Peak theoretical throughput by precision type
| Precision | Dense | 2:4 Sparse |
|---|---|---|
FP64 | No verified data available | Structured sparsity is a tensor-core feature; this vector precision has no sparse form |
FP32 32-bit floating point | 181TFLOPS | Structured sparsity is a tensor-core feature; this vector precision has no sparse form |
TF32 TensorFloat-32 | 667TFLOPS | 2,563TFLOPS |
BF16 Brain Float 16 | 667TFLOPS | 2,563TFLOPS |
FP16 16-bit floating point | 667TFLOPS | 2,563TFLOPS |
FP8 8-bit floating point | 1,299TFLOPS | 2,563TFLOPS |
FP6 | No verified data available | The part supports 2:4 sparsity, but the vendor publishes no sparse figure for this precision |
FP4 | No verified data available | The part supports 2:4 sparsity, but the vendor publishes no sparse figure for this precision |
INT8 | No verified data available | The part supports 2:4 sparsity, but the vendor publishes no sparse figure for this precision |
Specifications
Architecture
NeuronCore-v3
Form Factor
No verified data available
Launch Year
2024
Process Node
No verified data available
Memory
96 GB
Bandwidth
2,900 GB/s
TDP
No verified data available
Max power
No verified data available
Interconnect
1.3 TB/s NeuronLink-v3
direction and scope not stated by vendor
NeuronCore-v3
8
Spec Confidence
Official
Full Specifications
| Compute Engine | |
|---|---|
| NeuronCore-v3 | 8 |
| Memory | |
| Memory | 96 GB |
| Memory Type | No verified data available |
| Bandwidth | 2.9 TB/s |
| Interface Width | No verified data available |
| On-Die SRAM | 224 MB |
| Interconnect & I/O | |
| GPU-to-GPU | NeuronLink-v3 |
| Interconnect Bandwidth | 1.3 TB/s direction and scope not stated by vendor |
| Power & Thermal | |
| TDP | No verified data available |
| Max power | No verified data available |
| Enterprise Features | |
| Sparsity | Yes |
| General | |
| Form Factor | No verified data available |
| Architecture | NeuronCore-v3 |
| Process Node | No verified data available |
| Launch Year | 2024 |
Datasheet & Resources
Data Provenance
Every figure traced to a source
Primary Source
- Publisher
- AWS
- Published
- No verified data available
Data Quality
- Spec confidence
- Official
- Clock basis
- Boost
- Core precisions with figures
- 5 of 9
- Normalization
- All values in TFLOPS
Compare cloud providers offering on-demand GPU instances for AI training, inference, and HPC workloads.
Browse GPU Cloud ProvidersSystems Using This GPU
Pre-configured systems featuring the AWS Trainium2
| System | GPU Count | Peak Performance | Total Power | |
|---|---|---|---|---|
AWS EC2 trn2.48xlarge EC2 Instance · rack · 2024 | 16x Trainium2 | 20.8 PFLOPS | No verified data available | View System |
AWS Trn2 UltraServer UltraServer · rack · 2024 | 64x Trainium2 | 83.2 PFLOPS | No verified data available | View System |
Frequently Asked Questions
How many TFLOPS does the AWS Trainium2 have?
The AWS Trainium2 delivers 181 TFLOPS FP32, 667 TFLOPS FP16 and 1,299 TFLOPS FP8 at peak.
What is the power consumption of the AWS Trainium2?
Flopper does not currently have a verified TDP (Thermal Design Power) figure for the AWS Trainium2.
How much memory does the AWS Trainium2 have?
The AWS Trainium2 is equipped with 96 GB of memory with 2,900 GB/s of memory bandwidth.
What architecture is the AWS Trainium2 based on?
The AWS Trainium2 is based on the NeuronCore-v3 architecture, launched in 2024.
Stay Updated on GPU Releases
Get notified when new GPUs are added or specifications are updated.
No spam, unsubscribe anytime.