Closed yunlongxu-numagic closed 1 year ago
the gpu release has changed, we need to update the doc, and the colab notebook
pip install "jax[cuda11_cudnn82]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
the gpu release has changed, we need to update the doc, and the colab notebook