[jit] jax 高 动态 range gumbel 从 JIT 缓存密钥中缺失

However, the configuration state has include_in_trace_context=True but not include_in_jit_key=True:

use_high_dynamic_range_gumbel = bool_state(
name="jax_high_dynamic_range_gumbel",
default=False,
help="...",
include_in_trace_context=True,
)

Consequently, the two flag values produce different jaxprs, but the outer JIT cache can reuse an executable traced under the previous value.