jaxlib
jax[cpu]
