diff --git a/third_party/xla/xla/benchmarks/README.md b/third_party/xla/xla/benchmarks/README.md index fc4dd420d90639..128a1bbf807798 100644 --- a/third_party/xla/xla/benchmarks/README.md +++ b/third_party/xla/xla/benchmarks/README.md @@ -2,7 +2,10 @@ ## Prerequisites -- Python >= 3.12 (required by JAX >= 0.11) +These benchmarks are intended to run on a TPU v6e or 7x Cloud VM running Python +>= 3.12 (required by JAX >= 0.11). Please refer to +[these instructions](https://docs.cloud.google.com/tpu/docs/create-tpu-vm) for +steps on how to create one. ## Quickstart