jax>=0.4.0
jaxlib>=0.4.0
numpy>=1.22.0
scikit-image
tqdm>4.60.0
types-tqdm
typing-extensions

[cuda]
jax[cuda12]>=0.4.0

[dev]
mypy==2.1.0
pre-commit==4.6.0
pyright==1.1.409
pytest==9.0.3
ruff==0.15.14

[doc]
matplotlib>=3.5.0
numpy>=1.22.0
scipy>=1.10.0
sphinx>=7.4.0
sphinx-gallery>=0.17.0
sphinx-rtd-theme
