Tags: ff7250/jax
Tags
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
Merge pull request jax-ml#7793 from yashk2810:update_pypi PiperOrigin-RevId: 394697075
Merge pull request jax-ml#7774 from yashk2810:workspace PiperOrigin-RevId: 394233543
PreviousNext