numpy>=1.24
jax
jaxlib
