Align forward pass checker tolerances with Gemma3 standard patterns

Author: RexBearIU

tests/end_to_end/tpu/gpt_oss/20b/test_gpt_oss_to_mt.sh

@@ -54,5 +54,4 @@ python3 -m tests.utils.forward_pass_logit_checker \
     sparse_matmul=${SPARSE_MATMUL} \
     megablox=${MEGABLOX} \
     --golden_logits_path=/tmp/golden_data_gpt-oss-20b.jsonl \
-    --atol=0.1 --rtol=0.1 \
-    --max_kl_div=3e-4
+    --max_kl_div=0.01