GOOGLE
Cloud TPU v5p
459 BF16 · 95GB HBM2e · 250W
65Score
JAX TrainingGoogle Ecosystem
Specifications
ArchitectureTPU v5p
Memory95GB HBM2e
Memory Bandwidth2,765 GB/s
FP16 TFLOPS459
FP8 TFLOPS918
BF16 TFLOPS459
INT8 TOPS918
TDP250W
InterconnectICI (4,800 Gbps) (4800 GB/s)
EcosystemJAX
GenerationPrevious
Est. PriceCloud Only
Recommended Configuration
TPU v5p pod slice via GCP
Training Intelligence
◆JAX
◆TensorFlow
◆PyTorch/XLA
◆CUDA
◆ROCm
Training Time Estimates
LLaMA 70B(70B)
~18 days64 GPUs
GPT-3 175B(175B)
~25 days1024 GPUs
Stable Diffusion XL(3.5B)
~1.5 days8 GPUs
Cloud cost: $4.20/hr
Compare With
Ask AI Advisor