issue with 'pip install jax==0.3.4 jaxlib==0.3.2'

[forever208]

try pip install jax==0.3.25 jaxlib==0.3.25 if you have issue with the default installation