jax<1.0,>=0.9
numpy>=1.26
equinox>=0.13
diffrax>=0.7
einops>=0.8
optimistix>=0.0.9
scipy>=1.11
jaxtyping>=0.3

[all]
beamax[autodiff,dev,kwave,viz]

[autodiff]
optax>=0.2

[dev]
pytest
pytest-cov
ruff
pre-commit
pyright
h5py>=3.9
mkdocs
mkdocs-material
mkdocstrings-python

[kwave]
k-wave-python>=0.6.2

[viz]
matplotlib>=3.7
pyvista

[viz-mpl]
matplotlib>=3.7
