{"id":13533433,"url":"https://github.com/mpi4jax/mpi4jax","last_synced_at":"2026-02-21T19:34:30.717Z","repository":{"id":40249030,"uuid":"281372599","full_name":"mpi4jax/mpi4jax","owner":"mpi4jax","description":"Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:","archived":false,"fork":false,"pushed_at":"2026-02-19T13:07:15.000Z","size":5438,"stargazers_count":513,"open_issues_count":26,"forks_count":32,"subscribers_count":10,"default_branch":"main","last_synced_at":"2026-02-19T14:54:43.200Z","etag":null,"topics":["gpu","high-performance-computing","jax","jit","mpi","parallel-computing","xla"],"latest_commit_sha":null,"homepage":"https://mpi4jax.readthedocs.io/","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/mpi4jax.png","metadata":{"files":{"readme":"README.rst","changelog":null,"contributing":null,"funding":null,"license":"LICENSE.md","code_of_conduct":null,"threat_model":null,"audit":null,"citation":"CITATION.cff","codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null,"zenodo":null,"notice":null,"maintainers":null,"copyright":null,"agents":null,"dco":null,"cla":null}},"created_at":"2020-07-21T10:57:03.000Z","updated_at":"2026-02-06T15:07:50.000Z","dependencies_parsed_at":"2022-08-09T15:55:00.364Z","dependency_job_id":"ba5df72d-189a-4466-9e1b-d6bcf03304b7","html_url":"https://github.com/mpi4jax/mpi4jax","commit_stats":{"total_commits":260,"total_committers":6,"mean_commits":"43.333333333333336","dds":0.5769230769230769,"last_synced_commit":"5f4dfcbd02145dc7ad93389012dc62c53f6e81cd"},"previous_names":["philipvinc/mpi4jax"],"tags_count":87,"template":false,"template_full_name":null,"purl":"pkg:github/mpi4jax/mpi4jax","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/mpi4jax%2Fmpi4jax","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/mpi4jax%2Fmpi4jax/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/mpi4jax%2Fmpi4jax/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/mpi4jax%2Fmpi4jax/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/mpi4jax","download_url":"https://codeload.github.com/mpi4jax/mpi4jax/tar.gz/refs/heads/main","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/mpi4jax%2Fmpi4jax/sbom","scorecard":{"id":662860,"data":{"date":"2025-08-11","repo":{"name":"github.com/mpi4jax/mpi4jax","commit":"ba8e3a7985f90bb1b4fb48db6fdaf8d87a66da45"},"scorecard":{"version":"v5.2.1-40-gf6ed084d","commit":"f6ed084d17c9236477efd66e5b258b9d4cc7b389"},"score":5,"checks":[{"name":"Code-Review","score":5,"reason":"Found 9/17 approved changesets -- score normalized to 5","details":null,"documentation":{"short":"Determines if the project requires human code review before pull requests (aka merge requests) are merged.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#code-review"}},{"name":"Dangerous-Workflow","score":10,"reason":"no dangerous workflow patterns detected","details":null,"documentation":{"short":"Determines if the project's GitHub Action workflows avoid dangerous patterns.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#dangerous-workflow"}},{"name":"Maintained","score":3,"reason":"4 commit(s) and 0 issue activity found in the last 90 days -- score normalized to 3","details":null,"documentation":{"short":"Determines if the project is \"actively maintained\".","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#maintained"}},{"name":"CII-Best-Practices","score":0,"reason":"no effort to earn an OpenSSF best practices badge detected","details":null,"documentation":{"short":"Determines if the project has an OpenSSF (formerly CII) Best Practices Badge.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#cii-best-practices"}},{"name":"Security-Policy","score":0,"reason":"security policy file not detected","details":["Warn: no security policy file detected","Warn: no security file to analyze","Warn: no security file to analyze","Warn: no security file to analyze"],"documentation":{"short":"Determines if the project has published a security policy.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#security-policy"}},{"name":"Token-Permissions","score":0,"reason":"detected GitHub workflow tokens with excessive permissions","details":["Warn: no topLevel permission defined: .github/workflows/build-cuda-ext.yml:1","Warn: no topLevel permission defined: .github/workflows/build-docs.yml:1","Warn: no topLevel permission defined: .github/workflows/build-xpu-ext.yml:1","Warn: no topLevel permission defined: .github/workflows/covecov-coverage.yml:1","Warn: no topLevel permission defined: .github/workflows/mpi-tests.yml:1","Warn: no topLevel permission defined: .github/workflows/pre-commit.yml:1","Warn: no topLevel permission defined: .github/workflows/python-publish.yml:1","Info: no jobLevel write permissions found"],"documentation":{"short":"Determines if the project's workflows follow the principle of least privilege.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#token-permissions"}},{"name":"Vulnerabilities","score":10,"reason":"0 existing vulnerabilities detected","details":null,"documentation":{"short":"Determines if the project has open, known unfixed vulnerabilities.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#vulnerabilities"}},{"name":"Fuzzing","score":0,"reason":"project is not fuzzed","details":["Warn: no fuzzer integrations found"],"documentation":{"short":"Determines if the project uses fuzzing.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#fuzzing"}},{"name":"Binary-Artifacts","score":10,"reason":"no binaries found in the repo","details":null,"documentation":{"short":"Determines if the project has generated executable (binary) artifacts in the source repository.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#binary-artifacts"}},{"name":"License","score":10,"reason":"license file detected","details":["Info: project has a license file: LICENSE.md:0","Info: FSF or OSI recognized license: MIT License: LICENSE.md:0"],"documentation":{"short":"Determines if the project has defined a license.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#license"}},{"name":"Signed-Releases","score":-1,"reason":"no releases found","details":null,"documentation":{"short":"Determines if the project cryptographically signs release artifacts.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#signed-releases"}},{"name":"Branch-Protection","score":-1,"reason":"internal error: error during branchesHandler.setup: internal error: githubv4.Query: Resource not accessible by integration","details":null,"documentation":{"short":"Determines if the default and release branches are protected with GitHub's branch protection settings.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#branch-protection"}},{"name":"Pinned-Dependencies","score":0,"reason":"dependency not pinned by hash detected -- score normalized to 0","details":["Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/build-cuda-ext.yml:30: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/build-cuda-ext.yml/main?enable=pin","Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/build-cuda-ext.yml:37: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/build-cuda-ext.yml/main?enable=pin","Warn: third-party GitHubAction not pinned by hash: .github/workflows/build-cuda-ext.yml:64: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/build-cuda-ext.yml/main?enable=pin","Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/build-docs.yml:24: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/build-docs.yml/main?enable=pin","Warn: third-party GitHubAction not pinned by hash: .github/workflows/build-docs.yml:30: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/build-docs.yml/main?enable=pin","Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/build-xpu-ext.yml:18: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/build-xpu-ext.yml/main?enable=pin","Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/build-xpu-ext.yml:25: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/build-xpu-ext.yml/main?enable=pin","Warn: third-party GitHubAction not pinned by hash: .github/workflows/build-xpu-ext.yml:37: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/build-xpu-ext.yml/main?enable=pin","Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/covecov-coverage.yml:26: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/covecov-coverage.yml/main?enable=pin","Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/covecov-coverage.yml:33: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/covecov-coverage.yml/main?enable=pin","Warn: third-party GitHubAction not pinned by hash: .github/workflows/covecov-coverage.yml:38: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/covecov-coverage.yml/main?enable=pin","Warn: third-party GitHubAction not pinned by hash: .github/workflows/covecov-coverage.yml:61: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/covecov-coverage.yml/main?enable=pin","Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/mpi-tests.yml:39: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/mpi-tests.yml/main?enable=pin","Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/mpi-tests.yml:46: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/mpi-tests.yml/main?enable=pin","Warn: third-party GitHubAction not pinned by hash: .github/workflows/mpi-tests.yml:51: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/mpi-tests.yml/main?enable=pin","Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/pre-commit.yml:15: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/pre-commit.yml/main?enable=pin","Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/pre-commit.yml:16: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/pre-commit.yml/main?enable=pin","Warn: third-party GitHubAction not pinned by hash: .github/workflows/pre-commit.yml:17: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/pre-commit.yml/main?enable=pin","Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/python-publish.yml:20: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/python-publish.yml/main?enable=pin","Warn: GitHub-owned GitHubAction not pinned by hash: .github/workflows/python-publish.yml:27: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/python-publish.yml/main?enable=pin","Warn: third-party GitHubAction not pinned by hash: .github/workflows/python-publish.yml:32: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/python-publish.yml/main?enable=pin","Warn: third-party GitHubAction not pinned by hash: .github/workflows/python-publish.yml:46: update your workflow using https://app.stepsecurity.io/secureworkflow/mpi4jax/mpi4jax/python-publish.yml/main?enable=pin","Warn: pipCommand not pinned by hash: .github/workflows/build-cuda-ext.yml:47","Warn: pipCommand not pinned by hash: .github/workflows/build-cuda-ext.yml:48","Warn: pipCommand not pinned by hash: .github/workflows/build-cuda-ext.yml:49","Warn: pipCommand not pinned by hash: .github/workflows/build-cuda-ext.yml:50","Warn: pipCommand not pinned by hash: .github/workflows/build-cuda-ext.yml:70","Warn: pipCommand not pinned by hash: .github/workflows/build-cuda-ext.yml:71","Warn: pipCommand not pinned by hash: .github/workflows/build-xpu-ext.yml:43","Warn: pipCommand not pinned by hash: .github/workflows/build-xpu-ext.yml:44","Warn: pipCommand not pinned by hash: .github/workflows/covecov-coverage.yml:44","Warn: pipCommand not pinned by hash: .github/workflows/covecov-coverage.yml:45","Warn: pipCommand not pinned by hash: .github/workflows/covecov-coverage.yml:46","Warn: pipCommand not pinned by hash: .github/workflows/mpi-tests.yml:57","Warn: pipCommand not pinned by hash: .github/workflows/mpi-tests.yml:58","Warn: pipCommand not pinned by hash: .github/workflows/mpi-tests.yml:62","Warn: pipCommand not pinned by hash: .github/workflows/mpi-tests.yml:65","Warn: pipCommand not pinned by hash: .github/workflows/python-publish.yml:38","Warn: pipCommand not pinned by hash: .github/workflows/python-publish.yml:39","Info:   0 out of  13 GitHub-owned GitHubAction dependencies pinned","Info:   0 out of   9 third-party GitHubAction dependencies pinned","Info:   0 out of  17 pipCommand dependencies pinned"],"documentation":{"short":"Determines if the project has declared and pinned the dependencies of its build process.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#pinned-dependencies"}},{"name":"Packaging","score":10,"reason":"packaging workflow detected","details":["Info: Project packages its releases by way of GitHub Actions.: .github/workflows/python-publish.yml:11"],"documentation":{"short":"Determines if the project is published as a package that others can easily download, install, easily update, and uninstall.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#packaging"}},{"name":"SAST","score":0,"reason":"SAST tool is not run on all commits -- score normalized to 0","details":["Warn: 0 commits out of 29 are checked with a SAST tool"],"documentation":{"short":"Determines if the project uses static code analysis.","url":"https://github.com/ossf/scorecard/blob/f6ed084d17c9236477efd66e5b258b9d4cc7b389/docs/checks.md#sast"}}]},"last_synced_at":"2025-08-21T16:46:12.339Z","repository_id":40249030,"created_at":"2025-08-21T16:46:12.339Z","updated_at":"2025-08-21T16:46:12.339Z"},"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":286080680,"owners_count":29691045,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2026-02-21T18:18:25.093Z","status":"ssl_error","status_checked_at":"2026-02-21T18:18:22.435Z","response_time":107,"last_error":"SSL_read: unexpected eof while reading","robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":false,"can_crawl_api":true,"host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["gpu","high-performance-computing","jax","jit","mpi","parallel-computing","xla"],"created_at":"2024-08-01T07:01:19.803Z","updated_at":"2026-02-21T19:34:30.712Z","avatar_url":"https://github.com/mpi4jax.png","language":"Python","funding_links":[],"categories":["Software","Python","Libraries"],"sub_categories":["Trends"],"readme":"mpi4jax\n=======\n\n|JOSS paper| |PyPI Version| |Conda Version| |Tests| |codecov| |Documentation Status|\n\n``mpi4jax`` enables zero-copy, multi-host communication of `JAX \u003chttps://jax.readthedocs.io/\u003e`_ arrays, even from jitted code and from GPU memory.\n\n\nBut why?\n--------\n\nThe JAX framework `has great performance for scientific computing workloads \u003chttps://github.com/dionhaefner/pyhpc-benchmarks\u003e`_, but its `multi-host capabilities \u003chttps://jax.readthedocs.io/en/latest/jax.html#jax.pmap\u003e`_ are still limited.\n\nWith ``mpi4jax``, you can scale your JAX-based simulations to *entire CPU and GPU clusters* (without ever leaving ``jax.jit``).\n\nIn the spirit of differentiable programming, ``mpi4jax`` also supports differentiating through some MPI operations.\n\n\nInstallation\n------------\n\n``mpi4jax`` is available through ``pip`` and ``conda``:\n\n.. code:: bash\n\n   $ pip install mpi4jax                     # Pip\n   $ conda install -c conda-forge mpi4jax    # conda\n\nDepending on the different jax backends you want to use, you can install mpi4jax in the following way\n\n.. code:: bash\n\n   # pip install 'jax[cpu]'\n   $ pip install mpi4jax\n\n   # pip install -U 'jax[cuda12]'\n   $ pip install cython\n   $ pip install mpi4jax --no-build-isolation\n\n   # pip install -U 'jax[cuda12_local]'\n   $ CUDA_ROOT=XXX pip install mpi4jax\n\n(for more informations on jax GPU distributions, `see the JAX installation instructions \u003chttps://github.com/google/jax#installation\u003e`_)\n\nIn case your MPI installation is not detected correctly, `it can help to install mpi4py separately \u003chttps://mpi4py.readthedocs.io/en/stable/install.html\u003e`_. When using a pre-installed ``mpi4py``, you *must* use ``--no-build-isolation`` when installing ``mpi4jax``:\n\n.. code:: bash\n\n   # if mpi4py is already installed\n   $ pip install cython\n   $ pip install mpi4jax --no-build-isolation\n\n`Our documentation includes some more advanced installation examples. \u003chttps://mpi4jax.readthedocs.io/en/latest/installation.html\u003e`_\n\n\nExample usage\n-------------\n\n.. code:: python\n\n   from mpi4py import MPI\n   import jax\n   import jax.numpy as jnp\n   import mpi4jax\n\n   comm = MPI.COMM_WORLD\n   rank = comm.Get_rank()\n\n   @jax.jit\n   def foo(arr):\n      arr = arr + rank\n      arr_sum, _ = mpi4jax.allreduce(arr, op=MPI.SUM, comm=comm)\n      return arr_sum\n\n   a = jnp.zeros((3, 3))\n   result = foo(a)\n\n   if rank == 0:\n      print(result)\n\nRunning this script on 4 processes gives:\n\n.. code:: bash\n\n   $ mpirun -n 4 python example.py\n   [[6. 6. 6.]\n    [6. 6. 6.]\n    [6. 6. 6.]]\n\n``allreduce`` is just one example of the MPI primitives you can use. `See all supported operations here \u003chttps://mpi4jax.readthedocs.org/en/latest/api.html\u003e`_.\n\nCommunity guidelines\n--------------------\n\nIf you have a question or feature request, or want to report a bug, feel free to `open an issue \u003chttps://github.com/mpi4jax/mpi4jax/issues\u003e`_.\n\nWe welcome contributions of any kind `through pull requests \u003chttps://github.com/mpi4jax/mpi4jax/pulls\u003e`_. For information on running our tests, debugging, and contribution guidelines please `refer to the corresponding documentation page \u003chttps://mpi4jax.readthedocs.org/en/latest/developers.html\u003e`_.\n\nHow to cite\n-----------\n\nIf you use ``mpi4jax`` in your work, please consider citing the following article:\n\n::\n\n  @article{mpi4jax,\n    doi = {10.21105/joss.03419},\n    url = {https://doi.org/10.21105/joss.03419},\n    year = {2021},\n    publisher = {The Open Journal},\n    volume = {6},\n    number = {65},\n    pages = {3419},\n    author = {Dion Häfner and Filippo Vicentini},\n    title = {mpi4jax: Zero-copy MPI communication of JAX arrays},\n    journal = {Journal of Open Source Software}\n  }\n\n.. |Tests| image:: https://github.com/mpi4jax/mpi4jax/workflows/Tests/badge.svg\n   :target: https://github.com/mpi4jax/mpi4jax/actions?query=branch%3Amain\n.. |codecov| image:: https://codecov.io/gh/mpi4jax/mpi4jax/branch/main/graph/badge.svg\n   :target: https://codecov.io/gh/mpi4jax/mpi4jax\n.. |PyPI Version| image:: https://img.shields.io/pypi/v/mpi4jax\n   :target: https://pypi.org/project/mpi4jax/\n.. |Conda Version| image:: https://img.shields.io/conda/vn/conda-forge/mpi4jax.svg\n   :target: https://anaconda.org/conda-forge/mpi4jax\n.. |Documentation Status| image:: https://readthedocs.org/projects/mpi4jax/badge/?version=latest\n   :target: https://mpi4jax.readthedocs.io/en/latest/?badge=latest\n.. |JOSS paper| image:: https://joss.theoj.org/papers/10.21105/joss.03419/status.svg\n   :target: https://doi.org/10.21105/joss.03419\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fmpi4jax%2Fmpi4jax","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fmpi4jax%2Fmpi4jax","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fmpi4jax%2Fmpi4jax/lists"}