Remove custom hook that installs TPU deps by default (#1782)

Also modified symlnk-cudnn.sh script to auto-detect cuDNN major version

Author: gpupuck

.github/container/Dockerfile.maxtext

@@ -30,6 +30,12 @@ for pattern in \
     # tensorflow-cpu,tensorboard,tensorflow-text>=2.19.0 is incompatible with tensorflow==2.18.1
     sed -i "${pattern}" ${SRC_PATH_MAXTEXT}/dependencies/requirements/base_requirements/requirements.txt
 done
+
+# remove custom build hook that conflicts with our dependencies introduced in
+# https://github.com/AI-Hypercomputer/maxtext/commit/de886f9a918d17cfb025a353e7da2bc4d64c26b5
+sed -i -e '/^\[tool\.hatch\.build\.targets\.wheel\.hooks\.custom\]$/d' \
+       -e '/^path = "build_hooks\.py"$/d' \
+       ${SRC_PATH_MAXTEXT}/pyproject.toml
 EOF
 
 ###############################################################################

.github/container/symlnk-cudnn.sh

@@ -2,8 +2,6 @@
 
 set -ex
 
-CUDNN_MAJOR_VERSION=9
-
 # Create a prefix with include/ and lib/ directories containing symlinks to the cuDNN
 # version that was just installed; this is useful to pass to XLA to avoid it fetching
 # its own copy of cuDNN.
@@ -20,6 +18,14 @@ if [[ -z "${libcudnn_pkgs}" ]]; then
   exit 1
 fi
 
+# Auto-detect cuDNN major version from installed packages (e.g., libcudnn9, libcudnn10)
+CUDNN_MAJOR_VERSION=$(echo "${libcudnn_pkgs}" | grep -oP 'libcudnn\K\d+' | head -n1)
+if [[ -z "${CUDNN_MAJOR_VERSION}" ]]; then
+  echo "Could not determine CUDNN major version from packages: ${libcudnn_pkgs}"
+  exit 1
+fi
+echo "Detected CUDNN major version: ${CUDNN_MAJOR_VERSION}"
+
 for cudnn_file in $(dpkg -L ${libcudnn_pkgs} | sort -u); do
   # Skip unversioned symlinks since we'll create our own
   if [[ -f "${cudnn_file}" || -h "${cudnn_file}" ]]; then