jax
jaxlib
pint

[test]
pytest>=3.6
