GitHub / mpi4jax / mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
JSON API: http://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/mpi4jax%2Fmpi4jax
Stars: 476
Forks: 31
Open issues: 23
License: mit
Language: Python
Size: 5.06 MB
Dependencies parsed at: Pending
Created at: almost 5 years ago
Updated at: 15 days ago
Pushed at: 3 months ago
Last synced at: 8 days ago
Commit Stats
Commits: 260
Authors: 6
Mean commits per author: 43.33
Development Distribution Score: 0.577
More commit stats: https://commits.ecosyste.ms/hosts/GitHub/repositories/mpi4jax/mpi4jax
Topics: gpu, high-performance-computing, jax, jit, mpi, parallel-computing, xla