mpi4py>=3.0.1
numpy
jax>=0.5.1

[dev]
pytest>=6
pytest-cov>=2.10.1
coverage[toml]>=5
pre-commit
flake8==3.9.2
tqdm>=4.52

[dev:python_version >= "3.8.0"]
black==23.9
