equinox>=0.12.2
jax>=0.7.2
jaxtyping>=0.3.2
lineax
numpy>=2.3.1
scipy>=1.15.3

[gpu]
cupy-cuda12x>=13.6.0
jax[cuda12]>=0.7.2
nvmath-python[cu12]>=0.7.0
