jax>=0.6.2
numpy>=2.2.6
modax-solvers>=0.2.2
numba-enzyme-cuda>=0.3.0

[cuda12]
jax[cuda12]>=0.6.2

[cuda13]

[cuda13:python_version >= "3.11"]
jax[cuda13]>=0.9.1
