Merge branch 'main' into label

Author: Yadan-Wei

.gitignore

@@ -5,6 +5,7 @@ __pycache__
 .venv
 .ruff_cache
 .pytest_cache
+.vscode
 
 # Generated docs
 docs/index.md

.kiro/steering/docs.md

@@ -76,9 +76,9 @@ docs/src/
 
 - `constants.py` - Path constants, global variables, `GLOBAL_CONFIG`, `SITE_URL`, `README_PATH`, and `RELEASE_NOTES_REQUIRED_FIELDS`
 - `sorter.py` - Sorting tiebreaker functions: `platform_sorter`, `accelerator_sorter`, `repository_sorter`
-- `utils.py` - Utility functions: `load_yaml()`, `load_table_config()`, `load_jinja2()`, `render_table()`, `write_output()`, `parse_version()`, `clone_git_repository()`, `build_ecr_uri()`, `build_public_ecr_uri()`, `get_framework_order()`
+- `utils.py` - Utility functions: `load_yaml()`, `load_table_config()`, `load_jinja2()`, `render_table()`, `write_output()`, `parse_version()`, `clone_git_repository()`, `build_ecr_uri()`, `build_public_ecr_uri()`, `get_framework_order()`, `flatten_group_repos()`
 - `image_config.py` - `ImageConfig` class, image loaders (`load_repository_images`, `load_legacy_images`, `load_images_by_framework_group`), `sort_by_version`, `get_latest_image_uri`, `build_image_row`, `check_public_registry`
-- `generate.py` - `generate_index()`, `generate_support_policy()`, `generate_available_images()`, `generate_release_notes()`, `generate_all()`
+- `generate.py` - `generate_index()`, `generate_support_policy()`, `generate_available_images()`, `generate_release_notes()`, `generate_all()`, helpers: `_consolidate_framework_version()`, `_collapse_minor_versions()`
 - `macros.py` - MkDocs macros plugin integration
 - `hooks.py` - MkDocs hooks entry point
 
@@ -356,16 +356,19 @@ display_names:
   known_issues: "Known Issues"
 
 # Framework groups for support policy consolidation (lowercase keys)
+# Flat list: repos split directly to per-repo rows on mismatch
+# Nested dict: repos consolidate by sub-group first on mismatch
 framework_groups:
   pytorch:
-    - pytorch-training
-    - pytorch-inference
-    - pytorch-training-arm64
-    - pytorch-inference-arm64
-  tensorflow:
-    - tensorflow-training
-    - tensorflow-inference
-    - tensorflow-inference-arm64
+    pytorch-training:
+      - pytorch-training
+      - pytorch-training-arm64
+    pytorch-inference:
+      - pytorch-inference
+      - pytorch-inference-arm64
+  vllm:
+    - vllm
+    - vllm-arm64
 
 # Table order (controls order in available_images.md and support_policy.md)
 table_order:
@@ -376,20 +379,25 @@ table_order:
 
 ### Support Policy Consolidation
 
-The `framework_groups` configuration consolidates support policy rows by framework. Repositories in the same group are combined into a single row using the framework name (e.g., "PyTorch").
+The `framework_groups` configuration consolidates support policy rows by framework. The generation follows a 3-step flow:
 
-**Version Display:**
+**Validation (at load time):** `load_repository_images` validates that images sharing the same full version within a single repository have identical GA/EOP dates. Raises `ValueError` if not (this is a data bug).
 
-- Images with the same major.minor version (e.g., `2.6.0` and `2.6.1`) are consolidated into a single row displayed as `2.6` if they have identical GA/EOP dates
-- If patch versions have different GA/EOP dates, each is displayed separately with full version (e.g., `2.6.0`, `2.6.1`) and a warning is logged
+**Step 1 — Group by full version:** All images in a framework group are grouped by full version (e.g., `2.6.0`), deduplicated per repository (one representative image per repo since intra-repo consistency is guaranteed).
 
-**Requirements:**
+**Step 2 — Hierarchical consolidation** via `_consolidate_framework_version()`: For each full version, tries three levels of date agreement, stopping at the first that succeeds:
+
+1. Framework group — all repos agree → single row (e.g., "PyTorch")
+1. Sub-group — repos within a sub-group agree → one row per sub-group (e.g., "PyTorch Inference"). Only applies to nested dict groups in `framework_groups`.
+1. Per-repo — no agreement → one row per repository using its individual display name
+
+**Step 3 — Major.minor collapse:** Non-split entries are grouped by major.minor. If all full versions within a major.minor share the same dates, they collapse into a single row displayed as the major.minor (e.g., `2.6`). Collapse is skipped for any major.minor that has split (per-repo) rows.
+
+**Behavior:**
 
-- All repositories in a group that have a given full version (X.Y.Z) must have identical GA/EOP dates
 - Missing versions in some repositories are allowed (only present repos are consolidated)
-- A `ValueError` is raised if dates differ within a group for the same full version
 
-To add a new framework group, add an entry to `framework_groups` with the framework name as key and list of repositories as value.
+To add a new framework group, add an entry to `framework_groups` with the framework name as key and list of repositories as value. Use nested dict format if you need intermediate sub-group consolidation when dates differ.
 
 ### Reordering Tables and Columns
 

README.md

@@ -19,6 +19,12 @@ ______________________________________________________________________
 - **[2025/11/17]** Released first [SGLang DLCs](https://gallery.ecr.aws/deep-learning-containers/sglang)
   - SageMaker: `public.ecr.aws/deep-learning-containers/sglang:0.5.5-gpu-py312`
 
+### 📢 Support Updates
+
+- **[2026/02/10]** Extended support for PyTorch 2.6 Inference containers until June 30, 2026
+  - PyTorch 2.6 Inference images will continue to receive security patches and updates through end of June 2026
+  - For complete framework support timelines, see our [Support Policy](https://aws.github.io/deep-learning-containers/reference/support_policy/)
+
 ### 🎉 Hot Off the Press
 
 - 🌐

docs/.nav.yml

@@ -3,6 +3,7 @@ nav:
   - Getting Started:
     - get_started/index.md
     - Using Deep Learning Containers: get_started/using_dlcs.md
+    - Release Notifications: get_started/release_notifications.md
   - Release Notes:
     - releasenotes/index.md
     - Base: releasenotes/base/index.md

docs/DEVELOPMENT.md

@@ -81,10 +81,14 @@ eop: "2035-10-15"   # End of Patch date
 
 **Version Consolidation:**
 
-- Images with the same major.minor version (e.g., `2.6.0` and `2.6.1`) are consolidated into a single row displayed as `2.6` if they have identical GA/EOP dates
-- If patch versions have different GA/EOP dates, each is displayed separately with full version (e.g., `2.6.0`, `2.6.1`) and a warning is logged
+- Images with the same major.minor version and identical GA/EOP dates are consolidated into a single row displayed as `2.6` with the framework group name (e.g., "PyTorch")
+- If the same version has different GA/EOP dates across repository types (e.g., training vs inference), separate rows are created showing the specific repository type: "PyTorch Training" and "PyTorch Inference"
+- ARM64 variants are automatically consolidated with their base repository
+- If patch versions within the same repository have different GA/EOP dates, each is displayed separately with full version (e.g., `2.6.0`, `2.6.1`) and a warning is logged
 
-**Validation:** All images in the same framework group with the same full version (X.Y.Z) must have identical GA/EOP dates.
+**Flexibility:** Repositories in the same framework group (e.g., pytorch-training and pytorch-inference) can have different GA/EOP dates for the same version. The system will automatically create separate rows showing the specific repository type when dates differ.
+
+**Example:** If PyTorch 2.6 Training has EOP 2025-10-15 but PyTorch 2.6 Inference has EOP 2026-10-15, the support policy table will show two separate rows with "PyTorch Training" and "PyTorch Inference" in the Framework column.
 
 ---
 

docs/get_started/release_notifications.md

@@ -0,0 +1,35 @@
+# Receive Notifications on New Updates
+
+You can receive notifications whenever a new {{ dlc_short }} is released. Notifications are published with
+[Amazon Simple Notification Service ({{ sns }})](https://docs.aws.amazon.com/sns/latest/dg/welcome.html){:target="_blank"} using the following topic.
+
+```
+arn:aws:sns:us-west-2:767397762724:dlc-updates
+```
+
+Messages are posted here when a new {{ dlc_short }} is published. The version, metadata, and regional image URIs of the container will be included in
+the message.
+
+These messages can be received using several different methods. We recommend the following method.
+
+## Subscribe to the SNS Topic
+
+1. Open the [{{ sns }} console](https://console.aws.amazon.com/sns/home){:target="_blank"}.
+
+2. In the navigation bar, change the {{ aws }} Region to **US West (Oregon)**, if necessary. You must select the region where the {{ sns }}
+   notification that you are subscribing to was created.
+
+3. In the navigation pane, choose **Subscriptions**, **Create subscription**.
+
+4. For the **Create subscription** dialog box, do the following:
+
+   1. For **Topic ARN**, copy and paste the following Amazon Resource Name (ARN): `arn:aws:sns:us-west-2:767397762724:dlc-updates`
+
+   2. For **Protocol**, choose one from **[Amazon Simple Queue Service ({{ sqs }}), {{ lambda }}, Email, Email-JSON]**
+
+   3. For **Endpoint**, enter the email address or **Amazon Resource Name (ARN)** of the resource that you will use to receive the notifications.
+
+   4. Choose **Create subscription**.
+
+5. You receive a confirmation email with the subject line *{{ aws }} Notification - Subscription Confirmation*. Open the email and choose **Confirm
+   subscription** to complete your subscription.

docs/src/data/base/13.0.2-gpu-ec2.yml

@@ -0,0 +1,27 @@
+framework: Base
+version: "13.0.2"
+accelerator: gpu
+python: py313
+cuda: cu130
+os: ubuntu22.04
+platform: ec2
+ga: "2026-02-09"
+eop: "2027-02-09"
+public_registry: true
+
+tags:
+  - "13.0.2-gpu-py313-cu130-ubuntu22.04-ec2"
+  - "13.0.2-gpu-py313-ubuntu22.04-ec2"
+  - "13.0-gpu-py313-cu130-ubuntu22.04-ec2"
+  - "13.0-gpu-py313-ubuntu22.04-ec2"
+
+announcements:
+  - "Introduced Base containers with CUDA 13.0.2 for EC2, ECS, EKS"
+  - "Added Python 3.13 support"
+
+packages:
+  python: "3.13.11"
+  cuda: "13.0.2"
+  cudnn: "9.15.1.9"
+  nccl: "2.28.9-1"
+  efa: "1.47.0"

docs/src/data/pytorch-inference-arm64/2.6-cpu-ec2.yml

@@ -5,7 +5,7 @@ python: py312
 os: ubuntu22.04
 platform: ec2
 ga: "2025-01-29"
-eop: "2026-01-29"
+eop: "2026-06-30"
 public_registry: true
 
 tags:

docs/src/data/pytorch-inference-arm64/2.6-cpu-sagemaker.yml

@@ -5,7 +5,7 @@ python: py312
 os: ubuntu22.04
 platform: sagemaker
 ga: "2025-01-29"
-eop: "2026-01-29"
+eop: "2026-06-30"
 public_registry: true
 
 tags:

docs/src/data/pytorch-inference-arm64/2.6-gpu-ec2.yml

@@ -6,7 +6,7 @@ cuda: cu124
 os: ubuntu22.04
 platform: ec2
 ga: "2025-01-29"
-eop: "2026-01-29"
+eop: "2026-06-30"
 public_registry: true
 
 tags: