.flake8
.gitignore
.pre-commit-config.yaml
LICENSE.md
README.md
pyproject.toml
setup.py
.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
mpi4jax/__init__.py
mpi4jax/_version.py
mpi4jax/flush.py
mpi4jax/utils.py
mpi4jax/validation.py
mpi4jax/warn.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/collective_ops/__init__.py
mpi4jax/collective_ops/allreduce.py
mpi4jax/collective_ops/bcast.py
mpi4jax/collective_ops/recv.py
mpi4jax/collective_ops/send.py
mpi4jax/collective_ops/sendrecv.py
mpi4jax/cython/__init__.py
mpi4jax/cython/cuda_runtime_api.pxd
mpi4jax/cython/mpi_xla_bridge.c
mpi4jax/cython/mpi_xla_bridge.pxd
mpi4jax/cython/mpi_xla_bridge.pyx
mpi4jax/cython/mpi_xla_bridge_cpu.c
mpi4jax/cython/mpi_xla_bridge_cpu.pyx
mpi4jax/cython/mpi_xla_bridge_gpu.pyx
tests/test_allreduce_matvec.py
tests/test_collective_ops.py
tests/test_flush.py
tests/test_validation.py