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