jax>=0.8.0
numpy>=1.24.0

[compare]
SimpleITK>=2.3.0
matplotlib>=3.7
nibabel>=5.0

[cpu]
jax[cpu]>=0.8.0

[cuda12]
jax[cuda12]>=0.8.0

[dev]
pytest>=7.0
pytest-cov
ruff>=0.4.0
pre-commit>=3.0
SimpleITK>=2.3.0
