jax
numpy

[dev]
pytest
scipy
matplotlib
