MANIFEST.in
README.rst
pyproject.toml
setup.cfg
setup.py
versioneer.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/comm.py
mpi4jax/_src/decorators.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/mpi_xla_bridge.c
mpi4jax/_src/xla_bridge/mpi_xla_bridge_cpu.c