Resolved comments. Added deprecation warning and notes

Author: Rohan-Bierneni

docker_build_dependency_image.sh

@@ -21,6 +21,8 @@
 # (minutes). However, if you are simply changing local code and not updating dependencies, uploading just takes a few seconds.
 
 # bash docker_build_dependency_image.sh MODE=jax_ai_image BASEIMAGE={{JAX_AI_IMAGE BASEIMAGE FROM ARTIFACT REGISTRY}}
+# Note: The mode stable_stack is marked for deprecation, please use MODE=jax_ai_image instead
+# bash docker_build_dependency_image.sh MODE=stable_stack BASEIMAGE={{JAX_STABLE_STACK_IMAGE BASEIMAGE FROM ARTIFACT REGISTRY}}
 # bash docker_build_dependency_image.sh MODE=nightly
 # bash docker_build_dependency_image.sh MODE=stable JAX_VERSION=0.4.13
 # bash docker_build_dependency_image.sh MODE=stable

docs/getting_started/run_maxdiffusion_via_xpk.md

@@ -62,7 +62,7 @@ after which log out and log back in to the machine.
     bash docker_build_dependency_image.sh
     ```
 
-    #### New: Build MaxDiffusion Docker Image with JAX AI Images
+    #### New: Build MaxDiffusion Docker Image with JAX AI Images (Formerly known as JAX Stable Stack)
     We're excited to announce that you can build the MaxDiffusion Docker image using the JAX AI base image. This provides a more reliable and consistent build environment.
 
     ###### What is JAX AI Images?