search

conda-forge / jaxlib-feedstock

A conda-smithy repository for jaxlib.
BSD 3-Clause "New" or "Revised" License
16 stars 24 forks source link

Add Support For Windows? #161

Open adam-hartshorne opened 1 year ago

adam-hartshorne commented 1 year ago

Comment:

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

ngam commented 1 year ago

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

adam-hartshorne commented 1 year ago

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 commented 1 year ago

Related upstream issue: https://github.com/google/jax/issues/15907 .

hawkinsp commented 1 year ago

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 commented 1 year ago

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 commented 8 months ago

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.