Fix import from MaxText to maxtext

Author: hsuan-lun-chiang

src/MaxText/configs/base.yml

@@ -1035,7 +1035,7 @@ subslice_shape: ""
 
 # NNX
 enable_nnx: false
-pure_nnx_decoder: false
+pure_nnx_decoder: True
 
 ################################## Qwen3-Next Specific Configs ##################################
 # Kernel size for the 1D convolution in the Gated Delta Net

src/MaxText/layers/models.py

@@ -37,7 +37,6 @@
 from maxtext.multimodal import processor as mm_processor
 from maxtext.utils import max_utils
 
-
 # ------------------------------------------------------------------------------
 # The network: Transformer Definitions
 # ------------------------------------------------------------------------------

src/MaxText/layers/nnx_decoders.py

@@ -37,7 +37,6 @@
 from MaxText.layers import linears
 from MaxText.layers import initializers
 from MaxText.layers import quantizations
-from MaxText import maxtext_utils
 from MaxText import sharding
 from MaxText.layers.attentions import Attention
 from MaxText.layers.normalizations import RMSNorm