cloudhan

It seems that what we need is `actions.ptxas`, that is, a new rule that calls `ptxas` directly, instead of trying to mold it into `actions.compile`. If it is the case,...

Oops, I took it as generate from PTX. But OP is requiring generate to PTX. That is, we need to extend `actions.compile` to support produce ptx. But as well as...

Use `-f https://whls.blob.core.windows.net/unstable/index.html` may not work because jax changed their extras options handling. You need to open the link and download the whl file manually, and install the **compatible jax**...

Could you please set environment variable `TF_CPP_MIN_LOG_LEVEL` to 0? ```python import os os.environ["TF_CPP_MIN_LOG_LEVEL"] = "0" import jax jax.numpy.array([0]) ``` There used to be some useful dll info, not sure how...

`conda install nvidiatoolkit` does it config the PATH for you? If not, you might need to manually config the PATH to include the dir of cuda libraries.

Then does cudnn*.dll exists under that dir?

It is unmaintainable. The matched version pairs are selected only because the jax officially have that combination in their releases. But you should be able to build jaxlib from source...

This is not released, they are some side effect of [overhaul](https://github.com/cloudhan/jax-windows-builder/tree/overhaul) branch. Please don't use them at the moment.

The combination of the build is cuda 12.1 + cudnn 8.9. https://github.com/cloudhan/jax-windows-builder/blob/c53944319dd133592dfdd6593a5f73321fc45e83/build-jaxlib.ps1#L35-L42

Yes.