Fix pip dependency after cuDNN 9 upgrade

PiperOrigin-RevId: 637959308

Author: beckerhe

setup.py

@@ -22,7 +22,7 @@
 _current_jaxlib_version = '0.4.28'
 # The following should be updated with each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.4.28'
-_default_cuda12_cudnn_version = '89'
+_default_cuda12_cudnn_version = '91'
 _available_cuda12_cudnn_versions = [_default_cuda12_cudnn_version]
 _libtpu_version = '0.1.dev20240508'