Skip to content

Tags: GeraldCSC/jax

Tags

jax-v0.2.27

Toggle jax-v0.2.27's commit message
jax version 0.2.27

jax-v0.2.26

Toggle jax-v0.2.26's commit message
Update jaxlib version number to 0.1.76

PiperOrigin-RevId: 415050863

jaxlib-v0.1.75

Toggle jaxlib-v0.1.75's commit message
Merge pull request jax-ml#8842 from google:yashk2810-patch-3

PiperOrigin-RevId: 414715679

jaxlib-v0.1.74

Toggle jaxlib-v0.1.74's commit message
Jaxlib v0.1.74

jax-v0.2.25

Toggle jax-v0.2.25's commit message
jax version 0.2.25

jax-v0.2.24

Toggle jax-v0.2.24's commit message
Update jax version number for jax release.

PiperOrigin-RevId: 404262742

jax-v0.2.23

Toggle jax-v0.2.23's commit message
Add a default cuda installation path and more explicit installation p…

…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

jaxlib-v0.1.73

Toggle jaxlib-v0.1.73's commit message
Update the workspace file

PiperOrigin-RevId: 404076864

jax-v0.2.22

Toggle jax-v0.2.22's commit message
jax version 0.2.22

jaxlib-v0.1.72

Toggle jaxlib-v0.1.72's commit message
Merge pull request jax-ml#8181 from skye:workspace

PiperOrigin-RevId: 402632543