jax[cpu]
numpy

[dev]
pytest
nbdev
