Merge branch 'master' into patch-1

Author: dacorvo

.gitignore

@@ -26,6 +26,7 @@ doc/_static
 doc/_templates
 **/.DS_Store
 venv/
+.venv/
 *~
 .pytest_cache/
 *.swp

CHANGELOG.md

@@ -1,4 +1,26 @@
 # Changelog
+## v3.5.0 (2026-03-02)
+
+### Features
+- **Feature Store v3**: New version of Feature Store functionality
+- **Batch job listing by share identifier**: Added support for listing Batch jobs filtered by share identifier
+- **Stop condition for model customization trainers**: Added stopping condition support to model customization trainers
+- **EMRStep smart output**: Enhanced EMR step output handling with smart output capabilities
+- **Transform AMI version support**: Added support for specifying AMI version in SageMaker transform jobs
+
+### Enhancements
+- **Inference pipeline notebook example**: Added example notebook demonstrating inference pipeline usage
+- **Migration documentation**: Added migration documentation
+
+### Bug Fixes
+- **Model Customization bugs**: Fixed multiple issues in Model Customization functionality
+- **Default stopping condition removal**: Removed default stopping condition for MC trainer to prevent conflicts
+- **Instance groups parameter handling**: Fixed issue where default instance_type/instance_count were incorrectly applied when instance_groups was set
+- **JumpStart alt config resolution**: Resolved alternative configuration resolution for JumpStart models
+- **Inference processor naming**: Updated inference processor identifier from 'inf2' to 'neuronx'
+- **HuggingFace Neuronx PyTorch version**: Corrected the PyTorch version for HuggingFace Neuronx
+- **License additions**: Added license to sagemaker-mlops and sagemaker-serve packages
+
 ## v3.4.1 (2026-02-10)
 
 ### Fixes

VERSION

@@ -1 +1 @@
-3.4.1
+3.5.0

docs/api/sagemaker_mlops.rst

@@ -28,3 +28,12 @@ Local Development
    :members:
    :undoc-members:
    :show-inheritance:
+
+
+Feature Store
+-------------
+
+.. automodule:: sagemaker.mlops.feature_store
+   :members:
+   :undoc-members:
+   :show-inheritance:

pyproject.toml

@@ -33,10 +33,10 @@ classifiers = [
   "Programming Language :: Python :: 3.12",
 ]
 dependencies = [
-  "sagemaker-core>=2.4.1,<3.0.0",
-  "sagemaker-train>=1.4.1,<2.0.0",
-  "sagemaker-serve>=1.4.1,<2.0.0",
-  "sagemaker-mlops>=1.4.1,<2.0.0",
+  "sagemaker-core>=2.5.0,<3.0.0",
+  "sagemaker-train>=1.5.0,<2.0.0",
+  "sagemaker-serve>=1.5.0,<2.0.0",
+  "sagemaker-mlops>=1.5.0,<2.0.0",
 ]
 
 [project.optional-dependencies]

requirements/extras/test_requirements.txt

@@ -11,4 +11,5 @@ scipy
 omegaconf
 graphene
 typing_extensions>=4.9.0
-tensorflow>=2.16.2,<=2.19.0
+tensorflow>=2.16.2,<=2.19.0
+build

sagemaker-core/CHANGELOG.md

@@ -1,4 +1,16 @@
 # Changelog
+## v2.5.0 (2026-03-02)
+
+### Features
+
+- Feature Store V3 support (#5539)
+- Support transform AMI version in SageMaker transform jobs (#5521)
+
+### Bug Fixes
+
+- Update inference processor from 'inf2' to 'neuronx' (#5488)
+- Correct HF neuronx pytorch version (#5501)
+
 ## v2.4.1 (2026-02-10)
 
 ### Bug fixes and Other Changes

sagemaker-core/VERSION

@@ -1,2 +1,2 @@
-2.4.1
+2.5.0
 

sagemaker-core/src/sagemaker/core/common_utils.py

@@ -56,7 +56,9 @@
     "cn-north-1": "amazonaws.com.cn",
     "cn-northwest-1": "amazonaws.com.cn",
     "us-iso-east-1": "c2s.ic.gov",
+    "us-iso-west-1": "c2s.ic.gov",
     "us-isob-east-1": "sc2s.sgov.gov",
+    "us-isob-west-1": "sc2s.sgov.gov",
     "us-isof-south-1": "csp.hci.ic.gov",
     "us-isof-east-1": "csp.hci.ic.gov",
     "eu-isoe-west-1": "cloud.adc-e.uk",

sagemaker-core/src/sagemaker/core/image_uri_config/djl-lmi.json

@@ -46,7 +46,7 @@
                 "mx-central-1": "637423239942"
             },
             "repository": "djl-inference",
-            "tag_prefix": "0.36.0-lmi18.0.0-cu128"
+            "tag_prefix": "0.36.0-lmi21.0.0-cu129"
         },
         "0.35.0": {
             "registries": {
@@ -85,7 +85,13 @@
                 "ap-southeast-7": "590183813437",
                 "ap-southeast-4": "457447274322",
                 "ap-southeast-5": "550225433462",
-                "mx-central-1": "637423239942"
+                "mx-central-1": "637423239942",
+                "us-iso-east-1": "763104351884",
+                "us-iso-west-1": "763104351884",
+                "us-isob-east-1": "763104351884",
+                "us-isob-west-1": "763104351884",
+                "us-isof-east-1": "763104351884",
+                "us-isof-south-1": "763104351884"
             },
             "repository": "djl-inference",
             "tag_prefix": "0.35.0-lmi17.0.0-cu128"
@@ -127,7 +133,13 @@
                 "ap-southeast-7": "590183813437",
                 "ap-southeast-4": "457447274322",
                 "ap-southeast-5": "550225433462",
-                "mx-central-1": "637423239942"
+                "mx-central-1": "637423239942",
+                "us-iso-east-1": "763104351884",
+                "us-iso-west-1": "763104351884",
+                "us-isob-east-1": "763104351884",
+                "us-isob-west-1": "763104351884",
+                "us-isof-east-1": "763104351884",
+                "us-isof-south-1": "763104351884"
             },
             "repository": "djl-inference",
             "tag_prefix": "0.34.0-lmi16.0.0-cu128"
@@ -169,7 +181,13 @@
                 "ap-southeast-7": "590183813437",
                 "ap-southeast-4": "457447274322",
                 "ap-southeast-5": "550225433462",
-                "mx-central-1": "637423239942"
+                "mx-central-1": "637423239942",
+                "us-iso-east-1": "763104351884",
+                "us-iso-west-1": "763104351884",
+                "us-isob-east-1": "763104351884",
+                "us-isob-west-1": "763104351884",
+                "us-isof-east-1": "763104351884",
+                "us-isof-south-1": "763104351884"
             },
             "repository": "djl-inference",
             "tag_prefix": "0.33.0-lmi15.0.0-cu128"