From 1c57ea3c3311b23a2ed5d253a38e8e8b83d72d81 Mon Sep 17 00:00:00 2001 From: Weida Hong Date: Wed, 8 Apr 2026 05:03:56 +0000 Subject: [PATCH] Correct heading in README for TPU only topic Signed-off-by: Weida Hong --- CONTRIBUTING.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index cc08242..87d0aa7 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -19,7 +19,7 @@ pip install -e . pytest test ``` -## Setup on GPU or TPU +## TPU Setup Same as Mac setup, except, if you run test using pytest, please also add `JAX_PLATFORMS=cpu`. The reason is because pytest usually runs