Update rosetta/Dockerfile.maxtext

Co-authored-by: Terry Kong <[email protected]>

Author: ko3n1g

rosetta/Dockerfile.maxtext

@@ -1,5 +1,5 @@
 # syntax=docker/dockerfile:1-labs
-ARG BASE_IMAGE=ghcr.io/nvidia/jax-mealkit:upstream-pax
+ARG BASE_IMAGE=ghcr.io/nvidia/jax-mealkit:upstream-maxtext
 ARG [email protected]
 ARG GIT_USER_NAME=NVIDIA
 # If set to "true", then will pull new local patches, the manifest.yaml and create-distribution.sh (in case it was updated).