cloudhan/jax-windows-builder

TF Compression and TensorFlow Federated

madhavajay opened this issue · 3 comments

madhavajay commented

Now that these jaxlib wheels exist, the only compiled dependency for windows that prevents TFF from running is tensorflow-compression.

https://github.com/tensorflow/compression

If we can get the requirements for jax and jaxlib relaxed a little bit here:
google-parfait/tensorflow-federated#3242

I imagine TF compression is no where as complicated as jaxlib + cuda etc to get running in a fork and provide wheels.

Is this something you would be interested in some help with?

cloudhan commented

Sorry, not at the moment. Although it seems to be straightforward to maintain native builds when it builds, but it involves convoluted debugging process if errors happen.

madhavajay commented

@cloudhan okay no worries, maybe ill take a stab at it.
Any plans on a 0.3.17 release?

cloudhan commented

Added 0.3.17, see https://github.com/cloudhan/jax-windows-builder/actions/runs/3098881843

Note, that the jaxlib version ∈ (0.3.14, 0.3.17) is not buildable due to upstraem tf problem.

Close for now.

  • 👍1