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?
Jaxlib version
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.