jax>=0.4.38

[dev]
pytest>=8
scipy>=1.12
build>=1.2
twine>=6
