Add Support For Windows?

[adam-hartshorne]

Comment:

Would be great to add support for Windows version of jaxlib.

[ngam]

I agree it would be great to add support for Windows. I am not entirely sure where to begin though...

[adam-hartshorne]

This repository currently offers specific builds of Jax for windows, I presume a similar approach could be leveraged.

https://github.com/cloudhan/jax-windows-builder/

[traversaro]

Related upstream issue: jax-ml/jax#15907 .

[hawkinsp]

Note that upstream JAX now has Windows pip builds for CPU (only).

As to how you build such a wheel, see https://jax.readthedocs.io/en/latest/developer.html#additional-notes-for-building-jaxlib-from-source-on-windows and
https://github.com/google/jax/blob/main/.github/workflows/wheel_win_x64.yml
which is the pip wheel build.

[victoraalves]

Hi everyone,

Any news on this? as @hawkinsp mentioned, since 0.4.13 IIRC jax supports CPU-only on windows.

That would be really cool to have!

[ConorForgie]

I see on pypi they have it https://pypi.org/search/?q=jaxlib&o=&c=Operating+System+%3A%3A+Microsoft+%3A%3A+Windows - would be great to bring into the conda universe.