Merge branch 'main' into support

sirutBuasai · web-flow · commit b0dc93090c92 · 2026-02-10T17:21:49.000-08:00
diff --git a/.github/workflows/pr-sglang.yml b/.github/workflows/pr-sglang.yml
@@ -19,7 +19,7 @@ env:
   SGLANG_VERSION: "0.5.8"
   PYTHON_VERSION: "py312"
   CUDA_VERSION: "cu129"
-  OS_VERSION: "ubuntu22.04"
+  OS_VERSION: "ubuntu24.04"
 
   # SGLang EC2
   PROD_EC2_IMAGE: sglang:0.5-gpu-py312-ec2
diff --git a/.gitignore b/.gitignore
@@ -5,6 +5,7 @@ __pycache__
 .venv
 .ruff_cache
 .pytest_cache
+.vscode
 
 # Generated docs
 docs/index.md
diff --git a/.kiro/steering/docs.md b/.kiro/steering/docs.md
@@ -393,7 +393,7 @@ The `framework_groups` configuration consolidates support policy rows by framewo
 
 **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.
 
-**Requirements:**
+**Behavior:**
 
 - Missing versions in some repositories are allowed (only present repos are consolidated)
 
diff --git a/README.md b/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
 
 - 🌐
diff --git a/docs/DEVELOPMENT.md b/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.
 
 ---
 
diff --git a/docs/src/data/pytorch-inference-arm64/2.6-cpu-ec2.yml b/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-06-29"
+eop: "2026-06-30"
 public_registry: true
 
 tags:
diff --git a/docs/src/data/pytorch-inference-arm64/2.6-cpu-sagemaker.yml b/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-06-29"
+eop: "2026-06-30"
 public_registry: true
 
 tags:
diff --git a/docs/src/data/pytorch-inference-arm64/2.6-gpu-ec2.yml b/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-06-29"
+eop: "2026-06-30"
 public_registry: true
 
 tags:
diff --git a/docs/src/data/pytorch-inference/2.6-cpu-ec2.yml b/docs/src/data/pytorch-inference/2.6-cpu-ec2.yml
@@ -5,7 +5,7 @@ python: py312
 os: ubuntu22.04
 platform: ec2
 ga: "2025-01-29"
-eop: "2026-06-29"
+eop: "2026-06-30"
 public_registry: true
 
 tags:
diff --git a/docs/src/data/pytorch-inference/2.6-cpu-sagemaker.yml b/docs/src/data/pytorch-inference/2.6-cpu-sagemaker.yml
@@ -5,7 +5,7 @@ python: py312
 os: ubuntu22.04
 platform: sagemaker
 ga: "2025-01-29"
-eop: "2026-06-29"
+eop: "2026-06-30"
 public_registry: true
 
 tags:
diff --git a/docs/src/data/pytorch-inference/2.6-gpu-ec2.yml b/docs/src/data/pytorch-inference/2.6-gpu-ec2.yml
@@ -6,7 +6,7 @@ cuda: cu124
 os: ubuntu22.04
 platform: ec2
 ga: "2025-01-29"
-eop: "2026-06-29"
+eop: "2026-06-30"
 public_registry: true
 
 tags:
diff --git a/docs/src/data/pytorch-inference/2.6-gpu-sagemaker.yml b/docs/src/data/pytorch-inference/2.6-gpu-sagemaker.yml
@@ -6,7 +6,7 @@ cuda: cu124
 os: ubuntu22.04
 platform: sagemaker
 ga: "2025-01-29"
-eop: "2026-06-29"
+eop: "2026-06-30"
 public_registry: true
 
 tags:
diff --git a/docs/src/templates/reference/support_policy.template.md b/docs/src/templates/reference/support_policy.template.md
@@ -7,6 +7,10 @@
 - **GA (General Availability)**: The date when a framework version becomes officially supported and available for production use.
 - **EOP (End of Patch)**: The date after which a framework version no longer receives security patches or bug fixes.
 
+## Notice
+
+- We are extending support for PyTorch 2.6 Inference images until end of June 2026 as these are the last available PyTorch inference images with torchserve support.
+
 ## Supported Frameworks
 
 {{ supported_table }}
