Skip to content

Commit

Permalink
Merge pull request #20289 from parmentelat:patch-1
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 616880171
  • Loading branch information
jax authors committed Mar 18, 2024
2 parents 5a2fb38 + ed9a864 commit 0308aad
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/faq.rst
Original file line number Diff line number Diff line change
Expand Up @@ -337,7 +337,7 @@ referred to as being *sticky* to the device).
By default, JAX arrays are placed uncommitted on the default device
(``jax.devices()[0]``), which is the first GPU or TPU by default. If no GPU or
TPU is present, ``jax.devices()[0]`` is the CPU. The default device can
temporarily overridden with the :func:`jax.default_device` context manager, or
be temporarily overridden with the :func:`jax.default_device` context manager, or
set for the whole process by setting the environment variable ``JAX_PLATFORMS``
or the absl flag ``--jax_platforms`` to "cpu", "gpu", or "tpu"
(``JAX_PLATFORMS`` can also be a list of platforms, which determines which
Expand Down

0 comments on commit 0308aad

Please sign in to comment.