numpy
h5py
scipy
PyGnuplot
matplotlib
mpi4py

[:python_version < "3.8"]
importlib-metadata

[JAX]
jax
jaxlib

[MPI]
mpi4py

[MPI-JAX]
mpi4py
jax
jaxlib

[testing]
setuptools
pytest
pytest-cov
