fix jax & jaxlib installation #16

Closed xjli360 closed 10 months ago