jax>=0.11
numpy>=1.26

[test]
pytest
