avoid downgrading jax version
NONE
https://github.com/huggingface/datasets/blob/main/setup.py#L236
Currently has the highest version at 0.3.25; Not sure if there is any reason for this, other than that was the tested version?
2023-10-02T20:37:51Z
https://github.com/huggingface/datasets/issues/6202#issuecomment-1743728415
https://api.github.com/repos/huggingface/datasets/issues/6202
IC_kwDODunzps5n7y8f
https://api.github.com/repos/huggingface/datasets/issues/comments/1743728415/reactions
2023-10-02T20:37:51Z
https://api.github.com/repos/huggingface/datasets/issues/comments/1743728415
https://avatars.githubusercontent.com/u/4151002?v=4
https://api.github.com/users/markman123/events{/privacy}
https://api.github.com/users/markman123/followers
https://api.github.com/users/markman123/following{/other_user}
https://api.github.com/users/markman123/gists{/gist_id}
https://github.com/markman123
markman123
MDQ6VXNlcjQxNTEwMDI=
https://api.github.com/users/markman123/orgs
https://api.github.com/users/markman123/received_events
https://api.github.com/users/markman123/repos
https://api.github.com/users/markman123/starred{/owner}{/repo}
https://api.github.com/users/markman123/subscriptions
User
https://api.github.com/users/markman123
public
### Feature request

Whenever I `pip install datasets[jax]` it downgrades jax to version 0.3.25. I seem to be able to install this library first then upgrade jax back to version 0.4.13.

### Motivation

It would be nice to not overwrite currently installed version of jax if possible.

### Your contribution

I would be willing to beta test. Or maybe write some code if I could get pointed in the right direction, I'm not super familiar with this codebase.