Skip to content
@mpi4jax

mpi4jax

Pinned Loading

  1. mpi4jax mpi4jax Public

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

    Python 551 32

Repositories

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
    Python 551 MIT 32 19 (1 issue needs help) 6 Updated Sep 22, 2026
  • mpibackend4jax Public

    Easily use MPI as a backend for Jax native sharding

    mpi4jax/mpibackend4jax's past year of commit activity
    Python 3 0 1 1 Updated Jul 30, 2025
  • mpi4jax-joss Public

    JOSS paper for mpi4jax

    mpi4jax/mpi4jax-joss's past year of commit activity
    TeX 0 0 0 0 Updated Aug 31, 2021

Top languages

Python TeX

Most used topics

Loading…