diff --git a/phases/03-deep-learning-core/12-intro-to-jax/docs/en.md b/phases/03-deep-learning-core/12-intro-to-jax/docs/en.md index 078f7e20a5..ff91141e5d 100644 --- a/phases/03-deep-learning-core/12-intro-to-jax/docs/en.md +++ b/phases/03-deep-learning-core/12-intro-to-jax/docs/en.md @@ -438,13 +438,13 @@ pip install jax jaxlib optax flax For GPU support: ```bash -pip install jax[cuda12] +pip install 'jax[cuda12]' ``` For TPU (Google Cloud): ```bash -pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html +pip install 'jax[tpu]' -f https://storage.googleapis.com/jax-releases/libtpu_releases.html ``` **Performance gotchas:**