jax>=0.4.30
numba>=0.60.0

[cuda]
jax[cuda]>=0.6.2

[dev]
pytest>=8.4.1
pytest-benchmark>=5.1.0
pytype
black>=25.1.0
flake8>=7.3.0
flake8-pyproject>=1.2.3
pdoc>=15.0.4
mypy>=1.16.1
ipykernel>=6.29.5
matplotlib>=3.9.4

[notebook]
ipykernel>=6.29.5
matplotlib>=3.10.3
