jax
jaxlib

[benchmarking]
tmm-fast

[dev]
matplotlib
pytest
wheel
build
twine
sphinx
black
