numpy
scipy
scikit-learn
jax==0.7.0
jaxlib==0.7.0
jax-finufft
lineax

[dev]
pytest
pytest-cov
pytest-sugar
pre-commit
black
coveralls
ipython
matplotlib
