mpi4jax: Zero-copy MPI communication of JAX arrays
Dion Häfner, Filippo Vicentini
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.
Topics & Concepts
Zero (linguistics)Computer sciencePhilosophyLinguisticsAdvanced Data Compression TechniquesEmbedded Systems Design Techniques