[JAX] score_mod attention to support "enforce_precompiled" flag in cuDNN graph deserialization

[jberchtold-nvidia]

Once NVIDIA/cudnn-frontend#254 is completed, we should support this in TE's cuDNN usage with score_mod to give clearer error messages instead of potentially incorrect values or less clear errors in cases where the cuDNN Python FE and cuDNN C++ FE requires the cuDNN C++ FE to recompile the kernels.