Skip to content

Commit cf62573

Browse files
committed
Fix typo in multi_process.md
1 parent ae1cfc2 commit cf62573

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

docs/multi_process.md

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -28,11 +28,11 @@ Key concepts:
2828
* Each process has a
2929
distinct set of *local* devices it can address. The *global* devices are the set
3030
of all devices across all processes.
31-
* Use standard JAX parallelism APIs like {func}`~jax.pmap` and
32-
{func}`~jax.experimental.maps.xmap` . Each process “sees” *local* input and
31+
* Use standard JAX parallelism APIs like {func}`~jax.pmap` and
32+
{func}`~jax.experimental.maps.xmap`. Each process “sees” *local* input and
3333
output to parallelized functions, but communication inside the computations
3434
is *global*.
35-
* Make sure all processes run the same parallel computations in the same
35+
* Make sure all processes run the same parallel computations in the same
3636
order.
3737

3838
### Launching JAX processes
@@ -106,13 +106,13 @@ only launch computations on the 8 TPU cores attached directly to that host (see
106106
the
107107
[Cloud TPU System Architecture](https://cloud.google.com/tpu/docs/system-architecture)
108108
documentation for more details). You can see a process’s local devices via
109-
{func}`jax.local_devices()` .
109+
{func}`jax.local_devices()`.
110110

111111
**The *global* devices are the devices across all processes.** A computation can
112112
span devices across processes and perform collective operations via the direct
113113
communication links between devices, as long as each process launches the
114114
computation on its local devices. You can see all available global devices via
115-
{func}`jax.devices()` . A process’s local devices are always a subset of the
115+
{func}`jax.devices()`. A process’s local devices are always a subset of the
116116
global devices.
117117

118118
### Running multi-process computations

0 commit comments

Comments
 (0)