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?

Post not yet marked as solved Up vote post of PerEK Down vote post of PerEK
339 views
Posted by
PerEK
Reply
Add a Comment

Replies

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.

Posted by
steventrouble
Up vote reply of steventrouble Down vote reply of steventrouble
Add a Comment