.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/sharp-bits.rst
docs/usage.rst
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/warn.py
mpi4jax/_src/collective_ops/__init__.py
mpi4jax/_src/collective_ops/allreduce.py
mpi4jax/_src/collective_ops/bcast.py
mpi4jax/_src/collective_ops/recv.py
mpi4jax/_src/collective_ops/send.py
mpi4jax/_src/collective_ops/sendrecv.py
mpi4jax/_src/cython/__init__.py
mpi4jax/_src/cython/cuda_runtime_api.pxd
mpi4jax/_src/cython/mpi_xla_bridge.c
mpi4jax/_src/cython/mpi_xla_bridge.pxd
mpi4jax/_src/cython/mpi_xla_bridge.pyx
mpi4jax/_src/cython/mpi_xla_bridge_cpu.c
mpi4jax/_src/cython/mpi_xla_bridge_cpu.pyx
mpi4jax/_src/cython/mpi_xla_bridge_gpu.pyx
tests/test_allreduce_matvec.py
tests/test_collective_ops.py
tests/test_deprecations.py
tests/test_flush.py
tests/test_validation.py