Jaxlib version

I am trying hard to get some Whisper software running on Mac under jax. However, this requires jaxlib>=0.4.14. The current metal-jax requires jaxlib==0.4.11.

Anyone knows if there is any planned upgrade?

Agreed, I would love to use jax-metal, but it is half a year out of date and pip cannot find any version of flax that works with this old version.

Jaxlib version
 
 
Q