mpi4jax: Zero-copy MPI communication of JAX arrays

Abstract

The tensor framework JAX Expressivity is achieved by treating functions as first-class objects, while efficiency is obtained by compiling to machine code justahead-of-time. * Contributed equally, order determined by coin flip.