mpi4py>=3.0.1
numpy
jax>=0.6.0

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