numpy>=1.12
jax
jaxlib
