jax
jaxlib

[dev]
matplotlib
pytest
wheel
build
twine
sphinx
black
