Dirk Boonzajer

I fixed this in my version, you have to add ```package_dir={'': 'src'},``` to setup.py arguments. I submitted a pull request so will probably be fixed soon :). cheers! Dirk

Instructions to get halfway (python 3.10): ```shell pip install jaxlib==0.4.11+cuda12.cudnn89 -f https://whls.blob.core.windows.net/unstable/index.html --use-deprecated legacy-resolver # now, when importing we get # ModuleNotFoundError: No module named 'ml_dtypes._ml_dtypes_ext' # solve as per...

So, I downloaded the wheel, and installed ```pip install jax==0.4.13 ``` As far as I can see that should be compatible with jaxlib==0.4.11 (based on the source code) If I...

So, I reinstalled everything from scratch, just to make sure it's not because of some old environment that I tried: ```shell conda env create -n jax conda activate jax conda...

Once we get it to work, I could make a conda environment file that hopefully works without having to go through the same options. Would you be interested in including...

Yes, it does. I also just checked, looking with ``` import os os.environ ``` Gives, among others ```CUDA_PATH': 'C:\\ProgramData\\Anaconda3\\envs\\jax'``` I also installed cupy, that works without problems.