numpy>=2.1
ml_dtypes>=0.5.0
jaxlib==0.10.2

[dev]
pytest>=8
pytest-xdist>=3
pre-commit>=4
ruff>=0.15
build>=1.2
twine>=5
jupyter>=1
nbconvert>=7

[docs]
sphinx>=8
myst-parser>=4
alabaster>=1
sphinxcontrib-mermaid>=1

[gpu]
jaxlib==0.10.2
