mpi4jax
Results
1
repositories owned by
mpi4jax
mpi4jax
380
Stars
26
Forks
Watchers
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap: