jax>=0.11.1
numpy>=1.26

[cuda12]
jax[cuda12]>=0.11.1

[cuda13]
jax[cuda13]>=0.11.1

[dev]
pytest>=8

[plot]
matplotlib>=3.8
