Use-case guide · Text classification

Should you pick TPU v6e (Trillium) for text classification?

TPU v6e (Trillium) has 32 GB VRAM. Whether it's the right fit for text classification depends on your model size, expected QPS, and budget. Below is what we're seeing in production.

VRAM + model fit

TPU v6e (Trillium) fits models up to ~22B parameters in BF16 comfortably with room for KV-cache. For text classification specifically, you'll want to leave headroom for context length growth.

Pricing

Live pricing across all providers for TPU v6e (Trillium) is on the GPU detail page — click through for the sortable list.

Throughput

On text classification workloads, TPU v6e (Trillium) typically delivers the throughput published in its FP16 spec, minus the framework overhead (vLLM ≈ 85% MFU, TGI ≈ 70%).

Try the calculator to size the hardware for your specific model, or see all GPUs on the InferenceScore leaderboard.