Before using the steps below, you might want to check the issue jax-ml/jax#5795. I think if this has closed then
pip install jaxlib
should work on windows
This workaround is based on cloudhan/jax-windows-builder
Pre Install NOTE If you have a prior installation of jax remove it.
- Check if jax is installed using
pip show jax
- Uninstall using
pip uninstall jax
Steps for Installation