jax
jaxlib
scipy

[dev]
tox
