Add cuDNN FE patch for TE GKE build#2139

aybchan wants to merge 7 commits into

mainNVIDIA/JAX-Toolbox:mainfrom

aybchan/add-cudnn-fe-patch-gkeNVIDIA/JAX-Toolbox:aybchan/add-cudnn-fe-patch-gkeCopy head branch name to clipboard