Trainium2
other · other · 96 GB HBM2e · 600W TDP
VRAM
96 GB
BF16 TFLOPS
756
Bandwidth
3200 GB/s
From
$1.95/hr
Spec Sheet
Pricing by Provider
| Provider | On-Demand | Reserved | Spot | Badge |
|---|---|---|---|---|
| aws | $2.78/hr | $1.95/hr | - | Cheapest |
Compatible Models (280)
Single GPU (228 models)
Multi-GPU (52 models)
Training Capabilities
Estimated GPU count for full fine-tuning (AdamW, BF16) and QLoRA
| Model Size | Full Fine-Tune | QLoRA |
|---|---|---|
| 7B model | 2 GPUs | 1 GPU |
| 13B model | 3 GPUs | 1 GPU |
| 70B model | 14 GPUs | 1 GPU |
Energy Efficiency
Estimated tokens/second per Watt for popular models
Similar GPUs
| GPU | VRAM | BF16 TFLOPS | BW (GB/s) | From |
|---|---|---|---|---|
| Cloud AI 100 | 32 GB | 150 | 134 | $0.00/hr |
| Groq LPU | 230 GB | 188 | 80000 | $0.00/hr |
| H20 | 96 GB | 148 | 4000 | $0.99/hr |
| Gaudi 2 | 96 GB | 432 | 2460 | $1.65/hr |
| GH200 | 96 GB | 990 | 4000 | $2.99/hr |
Methodology Note
Performance estimates for the Trainium2are based on InferenceBench's roofline performance model with CUDA kernel-level optimization including FlashAttention v2 and PagedAttention. Memory calculations account for model weights (96 GB HBM2e available), KV-cache allocation, and activation memory. Throughput predictions use the Trainium2's rated 3200 GB/s memory bandwidth and 756 BF16 TFLOPS compute capacity as roofline ceilings, with empirical correction factors per GPU architecture (other). See our full methodology.
Frequently Asked Questions
How many AI models can run on Trainium2?
The Trainium2 can run 280 AI models from our database within a single node. Compatible models range across various parameter sizes depending on the quantization precision (BF16, FP8, INT4). Smaller models fit on a single GPU while larger models may require multi-GPU setups up to 16x Trainium2.
What is the Trainium2 inference throughput?
The Trainium2 delivers 756 BF16 TFLOPS and 1512 FP8 TFLOPS with 3200 GB/s memory bandwidth. Actual inference throughput (tokens/sec) depends on the model size, precision, and batch size. Use our calculator for model-specific throughput estimates.
How much does Trainium2 cost per hour?
The Trainium2 is available starting from $1.95/hour via aws. Prices vary by provider and pricing tier (on-demand, reserved, spot). Compare pricing across all providers in the table above.