Skip to content

jax-v0.2.23

jax-v0.2.23: Add a default cuda installation path and more explicit installation p…
@yashk2810 yashk2810 tagged this 19 Oct 02:56 
…aths for CUDA jaxlib.

```
# Installs Cuda 11 with Cudnn 8.2
$ pip install jax[cuda] -f https://storage.googleapis.com/jax-releases/jax_releases.html

$ pip install jax[cuda=11,cudnn=82] -f https://storage.googleapis.com/jax-releases/jax_releases.html

$ pip install jax[cuda=11,cudnn=805] -f https://storage.googleapis.com/jax-releases/jax_releases.html
```

PiperOrigin-RevId: 404134291
Assets 2