jax
jaxlib

[dev]
pytest
twine
