mpi4py>=3.0.1
numpy
jax>=0.3.25

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