Copybara import of the project:

--
de00614 by hsuan-lun-chiang <hsuan-lun.chiang@cienet.com>:

Migrate Gpt3 to NNX.

COPYBARA_INTEGRATE_REVIEW=#2062 from CIeNET-International:feat/Migrate-Gpt3-to-NNX de00614
PiperOrigin-RevId: 839048626

Author: hsuan-lun-chiang

src/MaxText/layers/decoders.py

@@ -412,7 +412,7 @@ def get_decoder_layers(self):
       case DecoderBlockType.GEMMA3:
         return [gemma3.Gemma3DecoderLayerToLinen]
       case DecoderBlockType.GPT3:
-        return [gpt3.Gpt3DecoderLayer]
+        return [gpt3.Gpt3DecoderLayerToLinen]
       case DecoderBlockType.GPT_OSS:
         return [gpt_oss.GptOssScannableBlockToLinen] if self.config.scan_layers else [gpt_oss.GptOssDecoderLayerToLinen]
       case DecoderBlockType.QWEN3:
@@ -598,7 +598,7 @@ def _apply_embedding(
           name="position_embedder",
           config=cfg,
           mesh=self.mesh,
-      )(decoder_positions, model_mode=model_mode)
+      )(decoder_positions.astype("int32"), model_mode=model_mode)
     return y
 
   @nn.compact