Downgrade the jaxlib requirement to match at least the one in public Colab and avoid https://github.com/google/jax/issues/2074
Downgrade the jaxlib requirement to match at least the one in public Colab and avoid https://github.com/google/jax/issues/2074

PiperOrigin-RevId: 291652219

google/neural-tangents