Skip to content

Commit 7ca1649

Browse files
committed
Update paper.md
1 parent cfb7f90 commit 7ca1649

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paper.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ The tensor framework JAX [@jax] combines expressivity and performance while prov
2828
In particular, JAX is expressive due to its clean, functional design, and performant due to a powerful JIT (just-in-time) compiler.
2929

3030
However, machine learning and (high-performance) scientific computing are often conducted on different hardware stacks: Machine learning is typically done on few highly parallel units (GPUs or TPUs) connected to a single host CPU, while scientific models tend to run on clusters of dozens to thousands of CPUs.
31-
Unfortunately, support from JAX and the underlying compiler XLA is much more mature in the former case.
31+
Unfortunately, support from JAX and the underlying compiler XLA is more mature in the former case.
3232
Notably, there is so far no built-in solution to communicate data between different nodes that is as sophisticated as the widely used MPI (Message Passing Interface) libraries [@mpistandard].
3333

3434
We attempt to fill this gap and introduce `mpi4jax`, a Python library bringing first-class support for the most important MPI operations to JAX.

0 commit comments

Comments
 (0)