added jaxlib specific version

[nubbthedestroyer]

In some cases, jaxlib will not install exact compatible version number. specifying version in requirements avoids startup error. I got this when running from PS on Windows 10.

The error message you're encountering, RuntimeError: jaxlib version 0.4.26 is newer than and incompatible with jax version 0.4.25. Please update your jax and/or jaxlib packages., indicates a version mismatch between jax and jaxlib. This is a common issue when the versions of these two packages are not compatible with each other. To resolve this issue, you need to ensure that both jax and jaxlib are updated to versions that are compatible.

[pafend]

For those looking to get this fixed fast, my solution was: pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html