Tags: GeraldCSC/jax
Tags
Update jaxlib version number to 0.1.76 PiperOrigin-RevId: 415050863
Merge pull request jax-ml#8842 from google:yashk2810-patch-3 PiperOrigin-RevId: 414715679
Update jax version number for jax release. PiperOrigin-RevId: 404262742
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
Update the workspace file PiperOrigin-RevId: 404076864
Merge pull request jax-ml#8181 from skye:workspace PiperOrigin-RevId: 402632543
PreviousNext