jax>=0.4
numpy

[reference]
scipy

[test]
scipy
pytest
