jax>=0.4.30
numpy

[dev]
pytest
scipy
