LaelaZorana/tpu-kernel-parity-lab-v5e download history

LaelaZorana/tpu-kernel-parity-lab-v5e is a dataset on the Hugging Face Hub. In the last 30 days it was downloaded 23 times (7 in the last 7 days), and 23 times in total. It ranks #338,096 among datasets by monthly downloads.

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 mu

Open LaelaZorana/tpu-kernel-parity-lab-v5e on Hugging Face

Sister project: Paper Pulse, the upvote history of every Hugging Face Daily Paper.