looking for the cuda11 type of jaxlib-0.3.15

[kevinycy]

I am looking for the cuda11 type of jaxlib-0.3.15,by https://storage.googleapis.com/jax-releases/libtpu_releases.html.I can only find the nocuda type. Thanks a lot!

[jakevdp]

You should be able to find it here: https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

pip install "jax[cuda]==0.3.15" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

[kevinycy]

thanks a lot!