Python
  • Blog
  • Contact

    • Xla Libraries

    1 Components & Libraries

    Sortby

    Xla Libraries

    mpi4jax
    Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:

    If you use pip and don't have JAX installed already, you will also need to do: Running this script on 4 processes gives:

    mpi4jax
    354 Mar 19, 2024

    Related tags

    Subscribe to our newsletter