Broken link to jax wheel in Colab notebook

[simonbatzner]

When I try running the colab notebooks, I get the following error in the first cell:

ERROR: HTTP error 404 while getting https://storage.googleapis.com/jax-wheels/cuda100/jaxlib-0.1.23-cp36-none-linux_x86_64.whl
  ERROR: Could not install requirement jaxlib==0.1.23 from https://storage.googleapis.com/jax-wheels/cuda100/jaxlib-0.1.23-cp36-none-linux_x86_64.whl because of error 404 Client Error: Not Found for url: https://storage.googleapis.com/jax-wheels/cuda100/jaxlib-0.1.23-cp36-none-linux_x86_64.whl
ERROR: Could not install requirement jaxlib==0.1.23 from https://storage.googleapis.com/jax-wheels/cuda100/jaxlib-0.1.23-cp36-none-linux_x86_64.whl because of HTTP error 404 Client Error: Not Found for url: https://storage.googleapis.com/jax-wheels/cuda100/jaxlib-0.1.23-cp36-none-linux_x86_64.whl for URL https://storage.googleapis.com/jax-wheels/cuda100/jaxlib-0.1.23-cp36-none-linux_x86_64.whl
     |████████████████████████████████| 215kB 1.4MB/s 
     |████████████████████████████████| 61kB 26.2MB/s 
  Building wheel for jax (setup.py) ... done
  Building wheel for opt-einsum (setup.py) ... done
  Building wheel for jax-md (setup.py) ... done

[sschoenholz]

Thanks for pointing this out! The notebook links have been updated. If you're interested (e.g. for your own notebooks), the reason why things broke is that JAX started storing the wheels at https://storage.googleapis.com/jax-releases/...

Let me know if you run into any issues!

[simonbatzner]

Works great now, thanks!