numpy>=1.25
jax>=0.5.0

[all]
warp-lang>=1.14
diffrax>=0.5

[cuda12]
jax[cuda12]
warp-lang>=1.14

[diffrax]
diffrax>=0.5

[docs]
sphinx
furo
myst-parser

[release]
build==1.5.0
twine==6.2.0
cffconvert==2.0.0
detect-secrets==1.5.0

[reproduce]
matplotlib
jupyter
nbconvert
pandas
scipy

[test]
pytest
scipy
ruff==0.16.5

[warp]
warp-lang>=1.14
