Trax Install on windows

Trying to install locally on windows machine. I get an error regarding dependency on jaxlib. Try to install jaxlib and get this. Any help to get around this or can this only be installed on linux os?

python -m pip install jaxlib

1 Like

Thanks for raising this.

JAX doesn’t support Windows right now, though it may be that the only thing standing in our way is getting the build to work. If Tensorflow can be built with XLA on Windows, then it should be possible to build JAX too.