.flake8
.git-archival.txt
.gitattributes
.gitignore
.pre-commit-config.yaml
.readthedocs.yml
LICENSE.md
README.rst
pyproject.toml
setup.py
.github/workflows/build-docs.yml
.github/workflows/covecov-coverage.yml
.github/workflows/mpi_tests.yml
.github/workflows/pre-commit.yml
.github/workflows/python-publish.yml
conf/ompi_rootenv.patch
conf/travis-install-mpi.sh
docs/Makefile
docs/api.rst
docs/conf.py
docs/environment.yml
docs/index.rst
docs/installation.rst
docs/make.bat
docs/shallow-water-source.rst
docs/shallow-water.rst
docs/sharp-bits.rst
docs/usage.rst
docs/_static/shallow-water.mp4
examples/shallow_water.py
mpi4jax/__init__.py
mpi4jax/_deprecations.py
mpi4jax/_version.py
mpi4jax.egg-info/PKG-INFO
mpi4jax.egg-info/SOURCES.txt
mpi4jax.egg-info/dependency_links.txt
mpi4jax.egg-info/requires.txt
mpi4jax.egg-info/top_level.txt
mpi4jax/_src/__init__.py
mpi4jax/_src/flush.py
mpi4jax/_src/utils.py
mpi4jax/_src/validation.py
mpi4jax/_src/collective_ops/__init__.py
mpi4jax/_src/collective_ops/allgather.py
mpi4jax/_src/collective_ops/allreduce.py
mpi4jax/_src/collective_ops/alltoall.py
mpi4jax/_src/collective_ops/bcast.py
mpi4jax/_src/collective_ops/gather.py
mpi4jax/_src/collective_ops/recv.py
mpi4jax/_src/collective_ops/reduce.py
mpi4jax/_src/collective_ops/scan.py
mpi4jax/_src/collective_ops/scatter.py
mpi4jax/_src/collective_ops/send.py
mpi4jax/_src/collective_ops/sendrecv.py
mpi4jax/_src/xla_bridge/__init__.py
mpi4jax/_src/xla_bridge/cuda_runtime_api.pxd
mpi4jax/_src/xla_bridge/mpi_xla_bridge.c
mpi4jax/_src/xla_bridge/mpi_xla_bridge.pxd
mpi4jax/_src/xla_bridge/mpi_xla_bridge.pyx
mpi4jax/_src/xla_bridge/mpi_xla_bridge_cpu.c
mpi4jax/_src/xla_bridge/mpi_xla_bridge_cpu.pyx
mpi4jax/_src/xla_bridge/mpi_xla_bridge_gpu.pyx
tests/conftest.py
tests/test_deprecations.py
tests/test_examples.py
tests/test_flush.py
tests/test_validation.py
tests/collective_ops/test_allgather.py
tests/collective_ops/test_allreduce.py
tests/collective_ops/test_allreduce_matvec.py
tests/collective_ops/test_alltoall.py
tests/collective_ops/test_bcast.py
tests/collective_ops/test_common.py
tests/collective_ops/test_gather.py
tests/collective_ops/test_reduce.py
tests/collective_ops/test_scan.py
tests/collective_ops/test_scatter.py
tests/collective_ops/test_send_and_recv.py
tests/collective_ops/test_sendrecv.py