tpu-v5e
Datasets
All datasets matching “tpu-v5e”runux-tpu-v5e-benchmarks
⚡ RunuX-AI — TPU v5e Inference Benchmarks
Achieving 3× Throughput & 3× Energy Reduction on Google TPU v5e
Xavier Callens · Socrate AI Lab (Non-Profit)
Reproducible benchmark data & scripts — No proprietary code included
🎯 What Is This?
This repository contains benchmark results and Apache-2.0 reproduction scripts for comparing LLM inference performance across 5 frameworks on Google TPU v5e. The goal is to enable independent verification of our claims… See the full description on the dataset page: https://huggingface.co/datasets/callensxavier/runux-tpu-v5e-benchmarks.tpu-kernel-parity-lab-v5e
JAX/Pallas attention measured on one TPU v5e device
I measured a Pallas masked-softmax kernel against the JAX/XLA reference inside grouped-query
attention. The Kaggle worker exposed eight TPU devices. The unsharded arrays ran on JAX's default
single device, so these numbers make no multi-device scaling claim.
All seven correctness checks passed. The benchmark contains twenty cases, with five warmups and
twenty synchronized timing samples in each case. XLA was faster in all ten… See the full description on the dataset page: https://huggingface.co/datasets/LaelaZorana/tpu-kernel-parity-lab-v5e.