mpi4jax: Zero-copy MPI communication of JAX arrays.

Dion Häfner Filippo Vicentini

Published in: J. Open Source Softw. (2021)

Keyphrases

multithreading
parallel computing
shared memory
distributed memory
message passing
parallel algorithm
high performance computing
multi agent systems
probabilistic model
parallel implementation
massively parallel
parallel programming
message passing interface
data mining
computing systems
parallel architectures