AI-Hypercomputer
diff --git a/‎src/maxtext/configs/base.yml‎
Lines changed: 2 additions & 2 deletions b/‎src/maxtext/configs/base.yml‎
Lines changed: 2 additions & 2 deletions
diff --git a/‎tests/unit/a_max_utils_test.py‎ ‎tests/a_unit/a_max_utils_test.py‎tests/unit/a_max_utils_test.py renamed to tests/a_unit/a_max_utils_test.py b/‎tests/unit/a_max_utils_test.py‎ ‎tests/a_unit/a_max_utils_test.py‎tests/unit/a_max_utils_test.py renamed to tests/a_unit/a_max_utils_test.py
diff --git a/‎tests/unit/attention_test.py‎ ‎tests/a_unit/attention_test.py‎tests/unit/attention_test.py renamed to tests/a_unit/attention_test.py b/‎tests/unit/attention_test.py‎ ‎tests/a_unit/attention_test.py‎tests/unit/attention_test.py renamed to tests/a_unit/attention_test.py
diff --git a/‎tests/unit/configs_test.py‎ ‎tests/a_unit/configs_test.py‎tests/unit/configs_test.py renamed to tests/a_unit/configs_test.py b/‎tests/unit/configs_test.py‎ ‎tests/a_unit/configs_test.py‎tests/unit/configs_test.py renamed to tests/a_unit/configs_test.py
diff --git a/‎tests/unit/configs_value_test.py‎ ‎tests/a_unit/configs_value_test.py‎tests/unit/configs_value_test.py renamed to tests/a_unit/configs_value_test.py b/‎tests/unit/configs_value_test.py‎ ‎tests/a_unit/configs_value_test.py‎tests/unit/configs_value_test.py renamed to tests/a_unit/configs_value_test.py
diff --git a/‎tests/unit/context_parallelism_test.py‎ ‎tests/a_unit/context_parallelism_test.py‎tests/unit/context_parallelism_test.py renamed to tests/a_unit/context_parallelism_test.py b/‎tests/unit/context_parallelism_test.py‎ ‎tests/a_unit/context_parallelism_test.py‎tests/unit/context_parallelism_test.py renamed to tests/a_unit/context_parallelism_test.py
diff --git a/‎tests/unit/data_loader_test.py‎ ‎tests/a_unit/data_loader_test.py‎tests/unit/data_loader_test.py renamed to tests/a_unit/data_loader_test.py b/‎tests/unit/data_loader_test.py‎ ‎tests/a_unit/data_loader_test.py‎tests/unit/data_loader_test.py renamed to tests/a_unit/data_loader_test.py
diff --git a/‎…sts/unit/deepseek32_vs_reference_test.py‎ ‎…s/a_unit/deepseek32_vs_reference_test.py‎tests/unit/deepseek32_vs_reference_test.py renamed to tests/a_unit/deepseek32_vs_reference_test.py b/‎…sts/unit/deepseek32_vs_reference_test.py‎ ‎…s/a_unit/deepseek32_vs_reference_test.py‎tests/unit/deepseek32_vs_reference_test.py renamed to tests/a_unit/deepseek32_vs_reference_test.py
diff --git a/‎tests/unit/deepseek_scan_engram_test.py‎ ‎tests/a_unit/deepseek_scan_engram_test.py‎tests/unit/deepseek_scan_engram_test.py renamed to tests/a_unit/deepseek_scan_engram_test.py b/‎tests/unit/deepseek_scan_engram_test.py‎ ‎tests/a_unit/deepseek_scan_engram_test.py‎tests/unit/deepseek_scan_engram_test.py renamed to tests/a_unit/deepseek_scan_engram_test.py
diff --git a/‎tests/unit/dequantize_mxfp4_test.py‎ ‎tests/a_unit/dequantize_mxfp4_test.py‎tests/unit/dequantize_mxfp4_test.py renamed to tests/a_unit/dequantize_mxfp4_test.py b/‎tests/unit/dequantize_mxfp4_test.py‎ ‎tests/a_unit/dequantize_mxfp4_test.py‎tests/unit/dequantize_mxfp4_test.py renamed to tests/a_unit/dequantize_mxfp4_test.py
@@ -1086,8 +1086,8 @@ position_id_per_seconds: 25
 subslice_shape: ""
 
 # NNX
-enable_nnx: false
-pure_nnx_decoder: false
+enable_nnx: True
+pure_nnx_decoder: True
 
 ################################## Qwen3-Next Specific Configs ##################################
 # Kernel size for the 1D convolution in the Gated Delta Net