update main and template

Signed-off-by: sirutBuasai <sirutbuasai27@outlook.com>

Author: sirutBuasai

docs/reference/available_images.template.md

@@ -1,15 +1,9 @@
 # Available Images
 
-This page is for referencing our supported deep learning containers.
+This page is for referencing our supported Deep Learning Containers.
 
 Refer to the tables below for all images that are available in AWS ECR repositories.
 
-## ECR Public Gallery
-
-Additionally, we also publish the following images to [ECR Public Gallery](https://gallery.ecr.aws/deep-learning-containers).
-
-{{ public_ecr_image_list }}
-
 ---
 
 ## Region Availability

docs/src/constants.py

@@ -13,3 +13,4 @@
 """Global variables for documentation generation."""
 
 AVAILABLE_IMAGES_TABLE_HEADER = "##"
+TUTORIALS_REPO = "https://github.com/aws-samples/sample-aws-deep-learning-containers"

docs/src/data/images.yml

@@ -5,25 +5,6 @@
 # Run `python docs/src/generate.py` to regenerate documentation from this file.
 # =============================================================================
 
-# -----------------------------------------------------------------------------
-# Public ECR Repositories
-# -----------------------------------------------------------------------------
-# List of repositories published to ECR Public Gallery
-# -----------------------------------------------------------------------------
-
-public_ecr_repositories:
-  - base
-  - sglang
-  - vllm
-  - vllm-arm64
-  - pytorch-training
-  - pytorch-inference
-  - pytorch-training-arm64
-  - pytorch-inference-arm64
-  - tensorflow-training
-  - tensorflow-inference
-  - tensorflow-inference-arm64
-
 # -----------------------------------------------------------------------------
 # Framework Support Policy
 # -----------------------------------------------------------------------------
@@ -118,7 +99,9 @@ support_policy:
 # -----------------------------------------------------------------------------
 # Container Images
 # -----------------------------------------------------------------------------
-# Organized by repository name. Tags are parsed to extract metadata.
+# Organized by repository name. Each repository contains:
+#   - public_registry: (optional) true if published to ECR Public Gallery
+#   - tags: list of image tags
 # Tag format: <version>-<processor>-<python>-[cuda]-<os>-<platform>
 # -----------------------------------------------------------------------------
 
@@ -127,183 +110,224 @@ images:
   # Base Containers
   # ===========================================================================
   base:
-    - "13.0.0-gpu-py312-cu130-ubuntu22.04-ec2"
-    - "12.9.1-gpu-py312-cu129-ubuntu22.04-ec2"
-    - "12.8.1-gpu-py312-cu128-ubuntu24.04-ec2"
-    - "12.8.0-gpu-py312-cu128-ubuntu22.04-ec2"
+    public_registry: true
+    tags:
+      - "13.0.0-gpu-py312-cu130-ubuntu22.04-ec2"
+      - "12.9.1-gpu-py312-cu129-ubuntu22.04-ec2"
+      - "12.8.1-gpu-py312-cu128-ubuntu24.04-ec2"
+      - "12.8.0-gpu-py312-cu128-ubuntu22.04-ec2"
 
   # ===========================================================================
   # vLLM Containers
   # ===========================================================================
   vllm:
-    - "0.13.0-gpu-py312-cu129-ubuntu22.04-sagemaker"
-    - "0.13.0-gpu-py312-cu129-ubuntu22.04-ec2"
+    public_registry: true
+    tags:
+      - "0.13.0-gpu-py312-cu129-ubuntu22.04-sagemaker"
+      - "0.13.0-gpu-py312-cu129-ubuntu22.04-ec2"
 
   # ===========================================================================
   # SGLang Containers
   # ===========================================================================
   sglang:
-    - "0.5.6-gpu-py312-cu129-ubuntu22.04-sagemaker"
+    public_registry: true
+    tags:
+      - "0.5.6-gpu-py312-cu129-ubuntu22.04-sagemaker"
 
   # ===========================================================================
   # PyTorch Containers
   # ===========================================================================
   pytorch-training:
-    - "2.9.0-gpu-py312-cu130-ubuntu22.04-sagemaker"
-    - "2.9.0-cpu-py312-ubuntu22.04-sagemaker"
-    - "2.9.0-gpu-py312-cu130-ubuntu22.04-ec2"
-    - "2.9.0-cpu-py312-ubuntu22.04-ec2"
-    - "2.8.0-gpu-py312-cu129-ubuntu22.04-sagemaker"
-    - "2.8.0-cpu-py312-ubuntu22.04-sagemaker"
-    - "2.8.0-gpu-py312-cu129-ubuntu22.04-ec2"
-    - "2.8.0-cpu-py312-ubuntu22.04-ec2"
-    - "2.7.1-gpu-py312-cu128-ubuntu22.04-sagemaker"
-    - "2.7.1-cpu-py312-ubuntu22.04-sagemaker"
-    - "2.7.1-gpu-py312-cu128-ubuntu22.04-ec2"
-    - "2.7.1-cpu-py312-ubuntu22.04-ec2"
-    - "2.6.0-gpu-py312-cu126-ubuntu22.04-sagemaker"
-    - "2.6.0-cpu-py312-ubuntu22.04-sagemaker"
-    - "2.6.0-gpu-py312-cu126-ubuntu22.04-ec2"
-    - "2.6.0-cpu-py312-ubuntu22.04-ec2"
+    public_registry: true
+    tags:
+      - "2.9.0-gpu-py312-cu130-ubuntu22.04-sagemaker"
+      - "2.9.0-cpu-py312-ubuntu22.04-sagemaker"
+      - "2.9.0-gpu-py312-cu130-ubuntu22.04-ec2"
+      - "2.9.0-cpu-py312-ubuntu22.04-ec2"
+      - "2.8.0-gpu-py312-cu129-ubuntu22.04-sagemaker"
+      - "2.8.0-cpu-py312-ubuntu22.04-sagemaker"
+      - "2.8.0-gpu-py312-cu129-ubuntu22.04-ec2"
+      - "2.8.0-cpu-py312-ubuntu22.04-ec2"
+      - "2.7.1-gpu-py312-cu128-ubuntu22.04-sagemaker"
+      - "2.7.1-cpu-py312-ubuntu22.04-sagemaker"
+      - "2.7.1-gpu-py312-cu128-ubuntu22.04-ec2"
+      - "2.7.1-cpu-py312-ubuntu22.04-ec2"
+      - "2.6.0-gpu-py312-cu126-ubuntu22.04-sagemaker"
+      - "2.6.0-cpu-py312-ubuntu22.04-sagemaker"
+      - "2.6.0-gpu-py312-cu126-ubuntu22.04-ec2"
+      - "2.6.0-cpu-py312-ubuntu22.04-ec2"
 
   pytorch-inference:
-    - "2.6.0-gpu-py312-cu124-ubuntu22.04-sagemaker"
-    - "2.6.0-cpu-py312-ubuntu22.04-sagemaker"
-    - "2.6.0-gpu-py312-cu124-ubuntu22.04-ec2"
-    - "2.6.0-cpu-py312-ubuntu22.04-ec2"
+    public_registry: true
+    tags:
+      - "2.6.0-gpu-py312-cu124-ubuntu22.04-sagemaker"
+      - "2.6.0-cpu-py312-ubuntu22.04-sagemaker"
+      - "2.6.0-gpu-py312-cu124-ubuntu22.04-ec2"
+      - "2.6.0-cpu-py312-ubuntu22.04-ec2"
 
   # ===========================================================================
   # TensorFlow Containers
   # ===========================================================================
   tensorflow-training:
-    - "2.19.0-gpu-py312-cu125-ubuntu22.04-sagemaker"
-    - "2.19.0-cpu-py312-ubuntu22.04-sagemaker"
+    public_registry: true
+    tags:
+      - "2.19.0-gpu-py312-cu125-ubuntu22.04-sagemaker"
+      - "2.19.0-cpu-py312-ubuntu22.04-sagemaker"
 
   tensorflow-inference:
-    - "2.19.0-gpu-py312-cu122-ubuntu22.04-sagemaker"
-    - "2.19.0-cpu-py312-ubuntu22.04-sagemaker"
+    public_registry: true
+    tags:
+      - "2.19.0-gpu-py312-cu122-ubuntu22.04-sagemaker"
+      - "2.19.0-cpu-py312-ubuntu22.04-sagemaker"
 
   # ===========================================================================
   # ARM64/Graviton Containers
   # ===========================================================================
   vllm-arm64:
-    - "0.10.2-gpu-py312-cu129-ubuntu22.04-ec2"
+    public_registry: true
+    tags:
+      - "0.10.2-gpu-py312-cu129-ubuntu22.04-ec2"
 
   pytorch-training-arm64:
-    - "2.7.0-gpu-py312-cu128-ubuntu22.04-ec2"
+    public_registry: true
+    tags:
+      - "2.7.0-gpu-py312-cu128-ubuntu22.04-ec2"
 
   pytorch-inference-arm64:
-    - "2.6.0-gpu-py312-cu124-ubuntu22.04-sagemaker"
-    - "2.6.0-gpu-py312-cu124-ubuntu22.04-ec2"
-    - "2.6.0-cpu-py312-ubuntu22.04-ec2"
+    public_registry: true
+    tags:
+      - "2.6.0-gpu-py312-cu124-ubuntu22.04-sagemaker"
+      - "2.6.0-gpu-py312-cu124-ubuntu22.04-ec2"
+      - "2.6.0-cpu-py312-ubuntu22.04-ec2"
+
+  tensorflow-inference-arm64:
+    public_registry: true
+    tags:
+      - "2.19.0-cpu-py312-ubuntu22.04-sagemaker"
 
   # ===========================================================================
   # Large Model Inference (LMI)
   # ===========================================================================
   djl-inference:
-    - "0.36.0-lmi18.0.0-cu128"
-    - "0.35.0-lmi17.0.0-cu128"
-    - "0.34.0-lmi16.0.0-cu128"
-    - "0.33.0-lmi15.0.0-cu128"
-    - "0.33.0-tensorrtllm0.21.0-cu128"
-    - "0.32.0-lmi14.0.0-cu126"
-    - "0.32.0-tensorrtllm0.12.0-cu125"
-    - "0.36.0-cpu-full"
-    - "0.35.0-cpu-full"
-    - "0.29.0-cpu-full"
+    tags:
+      - "0.36.0-lmi18.0.0-cu128"
+      - "0.35.0-lmi17.0.0-cu128"
+      - "0.34.0-lmi16.0.0-cu128"
+      - "0.33.0-lmi15.0.0-cu128"
+      - "0.33.0-tensorrtllm0.21.0-cu128"
+      - "0.32.0-lmi14.0.0-cu126"
+      - "0.32.0-tensorrtllm0.12.0-cu125"
+      - "0.36.0-cpu-full"
+      - "0.35.0-cpu-full"
+      - "0.29.0-cpu-full"
 
   # ===========================================================================
   # AutoGluon Containers
   # ===========================================================================
   autogluon-training:
-    - "1.4.0-gpu-py311-cu124-ubuntu22.04"
-    - "1.4.0-cpu-py311-ubuntu22.04"
-    - "1.3.0-gpu-py311-cu124-ubuntu22.04"
-    - "1.3.0-cpu-py311-ubuntu22.04"
+    tags:
+      - "1.4.0-gpu-py311-cu124-ubuntu22.04"
+      - "1.4.0-cpu-py311-ubuntu22.04"
+      - "1.3.0-gpu-py311-cu124-ubuntu22.04"
+      - "1.3.0-cpu-py311-ubuntu22.04"
 
   autogluon-inference:
-    - "1.4.0-gpu-py311-cu124-ubuntu22.04"
-    - "1.4.0-cpu-py311-ubuntu22.04"
-    - "1.3.0-gpu-py311-cu124-ubuntu22.04"
-    - "1.3.0-cpu-py311-ubuntu22.04"
+    tags:
+      - "1.4.0-gpu-py311-cu124-ubuntu22.04"
+      - "1.4.0-cpu-py311-ubuntu22.04"
+      - "1.3.0-gpu-py311-cu124-ubuntu22.04"
+      - "1.3.0-cpu-py311-ubuntu22.04"
 
   # ===========================================================================
   # HuggingFace Containers
   # ===========================================================================
   huggingface-pytorch-training:
-    - "2.5.1-transformers4.49.0-gpu-py311-cu124-ubuntu22.04"
-    - "2.1.0-transformers4.36.0-gpu-py310-cu121-ubuntu20.04"
-    - "2.0.0-transformers4.28.1-gpu-py310-cu118-ubuntu20.04"
-    - "1.13.1-transformers4.26.0-gpu-py39-cu117-ubuntu20.04"
+    tags:
+      - "2.5.1-transformers4.49.0-gpu-py311-cu124-ubuntu22.04"
+      - "2.1.0-transformers4.36.0-gpu-py310-cu121-ubuntu20.04"
+      - "2.0.0-transformers4.28.1-gpu-py310-cu118-ubuntu20.04"
+      - "1.13.1-transformers4.26.0-gpu-py39-cu117-ubuntu20.04"
 
   huggingface-pytorch-inference:
-    - "2.6.0-transformers4.49.0-gpu-py312-cu124-ubuntu22.04"
-    - "2.6.0-transformers4.49.0-cpu-py312-ubuntu22.04"
-    - "2.1.0-transformers4.37.0-gpu-py310-cu118-ubuntu20.04"
-    - "2.1.0-transformers4.37.0-cpu-py310-ubuntu22.04"
+    tags:
+      - "2.6.0-transformers4.49.0-gpu-py312-cu124-ubuntu22.04"
+      - "2.6.0-transformers4.49.0-cpu-py312-ubuntu22.04"
+      - "2.1.0-transformers4.37.0-gpu-py310-cu118-ubuntu20.04"
+      - "2.1.0-transformers4.37.0-cpu-py310-ubuntu22.04"
 
   huggingface-tensorflow-training:
-    - "2.6.3-transformers4.17.0-gpu-py38-cu112-ubuntu20.04"
+    tags:
+      - "2.6.3-transformers4.17.0-gpu-py38-cu112-ubuntu20.04"
 
   huggingface-tensorflow-inference:
-    - "2.11.1-transformers4.26.0-gpu-py39-cu112-ubuntu20.04"
-    - "2.11.1-transformers4.26.0-cpu-py39-ubuntu20.04"
+    tags:
+      - "2.11.1-transformers4.26.0-gpu-py39-cu112-ubuntu20.04"
+      - "2.11.1-transformers4.26.0-cpu-py39-ubuntu20.04"
 
   # ===========================================================================
   # HuggingFace Neuron Containers
   # ===========================================================================
   huggingface-pytorch-inference-neuronx:
-    - "2.8.0-transformers4.55.4-neuronx-py310-sdk2.26.0-ubuntu22.04"
-    - "2.7.1-transformers4.51.3-neuronx-py310-sdk2.24.1-ubuntu22.04"
-    - "2.1.2-transformers4.43.2-neuronx-py310-sdk2.20.0-ubuntu20.04"
+    tags:
+      - "2.8.0-transformers4.55.4-neuronx-py310-sdk2.26.0-ubuntu22.04"
+      - "2.7.1-transformers4.51.3-neuronx-py310-sdk2.24.1-ubuntu22.04"
+      - "2.1.2-transformers4.43.2-neuronx-py310-sdk2.20.0-ubuntu20.04"
 
   huggingface-pytorch-training-neuronx:
-    - "2.8.0-transformers4.55.4-neuronx-py310-sdk2.26.0-ubuntu22.04"
-    - "2.7.0-transformers4.51.0-neuronx-py310-sdk2.24.1-ubuntu22.04"
-    - "2.1.2-transformers4.48.1-neuronx-py310-sdk2.20.0-ubuntu20.04"
+    tags:
+      - "2.8.0-transformers4.55.4-neuronx-py310-sdk2.26.0-ubuntu22.04"
+      - "2.7.0-transformers4.51.0-neuronx-py310-sdk2.24.1-ubuntu22.04"
+      - "2.1.2-transformers4.48.1-neuronx-py310-sdk2.20.0-ubuntu20.04"
 
   # ===========================================================================
   # Neuron Containers
   # ===========================================================================
   pytorch-inference-neuronx:
-    - "2.8.0-neuronx-py311-sdk2.26.1-ubuntu22.04"
-    - "2.7.0-neuronx-py310-sdk2.25.0-ubuntu22.04"
-    - "2.7.0-neuronx-py310-sdk2.24.1-ubuntu22.04"
-    - "2.6.0-neuronx-py310-sdk2.23.0-ubuntu22.04"
+    tags:
+      - "2.8.0-neuronx-py311-sdk2.26.1-ubuntu22.04"
+      - "2.7.0-neuronx-py310-sdk2.25.0-ubuntu22.04"
+      - "2.7.0-neuronx-py310-sdk2.24.1-ubuntu22.04"
+      - "2.6.0-neuronx-py310-sdk2.23.0-ubuntu22.04"
 
   pytorch-training-neuronx:
-    - "2.8.0-neuronx-py311-sdk2.26.1-ubuntu22.04"
-    - "2.7.0-neuronx-py310-sdk2.25.0-ubuntu22.04"
-    - "2.7.0-neuronx-py310-sdk2.24.1-ubuntu22.04"
-    - "2.6.0-neuronx-py310-sdk2.23.0-ubuntu22.04"
+    tags:
+      - "2.8.0-neuronx-py311-sdk2.26.1-ubuntu22.04"
+      - "2.7.0-neuronx-py310-sdk2.25.0-ubuntu22.04"
+      - "2.7.0-neuronx-py310-sdk2.24.1-ubuntu22.04"
+      - "2.6.0-neuronx-py310-sdk2.23.0-ubuntu22.04"
 
   tensorflow-inference-neuronx:
-    - "2.10.1-neuronx-py310-sdk2.17.0-ubuntu20.04"
+    tags:
+      - "2.10.1-neuronx-py310-sdk2.17.0-ubuntu20.04"
 
   # ===========================================================================
   # StabilityAI Containers
   # ===========================================================================
   stabilityai-pytorch-inference:
-    - "2.0.1-sgm0.1.0-gpu-py310-cu118-ubuntu20.04-sagemaker"
+    tags:
+      - "2.0.1-sgm0.1.0-gpu-py310-cu118-ubuntu20.04-sagemaker"
 
   # ===========================================================================
   # SageMaker Training Compiler Containers
   # ===========================================================================
   pytorch-trcomp-training:
-    - "1.13.1-gpu-py39-cu117-ubuntu20.04-sagemaker"
-    - "1.12.0-gpu-py38-cu113-ubuntu20.04-sagemaker"
+    tags:
+      - "1.13.1-gpu-py39-cu117-ubuntu20.04-sagemaker"
+      - "1.12.0-gpu-py38-cu113-ubuntu20.04-sagemaker"
 
   huggingface-pytorch-trcomp-training:
-    - "1.11.0-transformers4.21.1-gpu-py38-cu113-ubuntu20.04"
+    tags:
+      - "1.11.0-transformers4.21.1-gpu-py38-cu113-ubuntu20.04"
 
   # ===========================================================================
   # NVIDIA Triton Containers
   # ===========================================================================
   sagemaker-tritonserver:
-    - "25.09-py3"
-    - "25.04-py3"
-    - "24.09-py3"
-    - "24.05-py3"
-    - "24.03-py3"
-    - "24.01-py3"
-    - "23.12-py3"
+    tags:
+      - "25.09-py3"
+      - "25.04-py3"
+      - "24.09-py3"
+      - "24.05-py3"
+      - "24.03-py3"
+      - "24.01-py3"
+      - "23.12-py3"

docs/src/hooks.py

@@ -18,6 +18,7 @@
 
 import os
 
+from constants import TUTORIALS_REPO
 from generate import generate_all
 from utils import clone_git_repository, load_yaml
 
@@ -32,7 +33,6 @@
 def on_startup(command=["build", "gh-deploy", "serve"], dirty=False):
     """MkDocs hook - runs before build."""
     yaml_data = load_yaml(DATA_FILE)
-    tutorials_repo = "https://github.com/aws-samples/sample-aws-deep-learning-containers"
 
-    clone_git_repository(tutorials_repo, TUTORIALS_DIR)
+    clone_git_repository(TUTORIALS_REPO, TUTORIALS_DIR)
     generate_all(yaml_data, dry_run=False)