jax>=0.4.30

[test]
pytest>=7
