Home Original page

mpi4jax

Pinned Loading

  1. Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡

    Python 517 32

Repositories

Type
Select type

Language
Select language

Sort
Select order

Showing 3 of 3 repositories

  • mpi4jax Public

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

    mpi4jax/mpi4jax’s past year of commit activity

  • mpibackend4jax Public

    Easily use MPI as a backend for Jax native sharding

    mpi4jax/mpibackend4jax’s past year of commit activity

    Python

    1

    0

    1 1

    Updated Jul 30, 2025

  • mpi4jax/mpi4jax-joss’s past year of commit activity

    TeX 0 0

    0 0

    Updated Aug 31, 2021