jax[cpu]==0.12.0.1
