Reply to jax-metal fails to install on M1 clean environment The jax-metal pypi package should set its dependencies to jax==0.4.26 jaxlib==0.4.26 rather than using >=. The devs can choose to patch the pypi release with a post release version. Machine Learning & AI General Aug ’24