Not installable on arm64 due to jaxlib upper bound

[paveldikov]

jaxlib does not ship source distributions on PyPI, i.e. is only available as a wheel.

Newer versions of jaxlib do provide aarch64 wheels, but transformers constrains jaxlib to <=0.4.13

Is it possible to relax this upper bound constraint? Or is there some very specific API-level breakage that was being guarded against? (I tried to infer by trawling through the git blame but nothing obvious jumped out at me -- looked like general defensiveness more than anything specific)