Llama 3 8B at FP16 needs 16.1 GB for its weights and 24.6 GB in total at 8K context with 8 concurrent users. That means it fits on a single TPU v6e Trillium (32GB). Every figure is computed by tokcalc's open-source capacity model.
| GPUs | Total VRAM | VRAM needed | Decode tok/s | Aggregate tok/s | TTFT | Cost / 1M tok | Fits? |
|---|---|---|---|---|---|---|---|
| 1× TPU | 32 GB | 24.6 GB | 33.15 | 397.8 | 346.8 ms | $0.5587 | ✓ |
| 2× TPU | 64 GB | 24.6 GB | 56.35 | 1352 | 191.1 ms | $0.3286 | ✓ |
| 4× TPU | 128 GB | 24.6 GB | 112.7 | 5410 | 110.5 ms | $0.1643 | ✓ |
| 8× TPU | 256 GB | 24.6 GB | 225.4 | 21639 | 70.3 ms | $0.0822 | ✓ |
| Context | Concurrent users | KV per request | Fits? |
|---|---|---|---|
| 4K | 29 | 549.8 MB | ✓ fits |
| 8K | 14 | 1.07 GB | ✓ fits |
| 33K | 3 | 4.29 GB | ✓ fits |
| 66K | 1 | 8.59 GB | ✓ fits |
| 131K | 0 | 17.18 GB | ✗ does not fit |
| GPU | GPUs | Aggregate tok/s | Cost / 1M tok |
|---|---|---|---|
| RTX 5090 32GB | 1× | 870.3 | $0.1596 |
| AMD MI300X 192GB | 1× | 2574 | $0.2698 |
| V100 SXM2 32GB | 1× | 437.1 | $0.3177 |
| A100 40GB SXM4 | 1× | 752.8 | $0.3321 |
| AMD MI325X 288GB | 1× | 2914 | $0.3336 |
| A100 80GB SXM4 | 1× | 971.4 | $0.3432 |
Yes. Llama 3 8B at FP16 needs 24.6 GB including the KV cache, and TPU v6e Trillium (32GB) has 32 GB — so it fits on a single GPU.
33.15 tokens/sec per stream and 397.8 tok/s aggregate with continuous batching at 8 concurrent users. Time to first token is 346.8 ms, and inter-token latency is 30.2 ms.
$0.5587 per 1M output tokens at 1× TPU v6e Trillium (32GB) ($0.8000/hr). That is the marginal hardware cost — add idle capacity, ops, and egress for the real figure.
16.1 GB for the weights at FP16, plus 8.6 GB of KV cache for 8 concurrent requests at 8K context — 24.6 GB total. Every extra concurrent user at long context costs more, so leave headroom.
It works. Compare $0.5587 per 1M tokens against the alternatives below — the cheapest option that fits the same workload is 4× RTX 3080 10GB at $0.0443, and the fastest is 1× Cerebras CS-3 (WSE-3) at 9714 tok/s.
Same formulas, your context length, concurrency, and quantisation.