jax<0.4.38,>=0.4.33

[build]
build>=1.2.2.post1
twine>=6.1.0

[test]
pytest>=8.3.4
