Jax does not specifically list jaxlib in its dependencies due to its various hardware-specific versions (such as for CPU, GPU, TPU, and so on). Python's dependency management system lacks the necessary precision to handle this. This issue is resolved by JAX through its installation guidelines.
Removing
jaxlib
from the dependencies.Jax does not specifically list
jaxlib
in its dependencies due to its various hardware-specific versions (such as for CPU, GPU, TPU, and so on). Python's dependency management system lacks the necessary precision to handle this. This issue is resolved by JAX through its installation guidelines.For more info: https://github.com/google/jax/discussions/16380