Skip to content

Commit 106a1ec

Browse files
authored
Merge branch 'master' into runtime_env_fix
2 parents c0766cf + 462bed0 commit 106a1ec

File tree

311 files changed

+118633
-30153
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

311 files changed

+118633
-30153
lines changed

.github/workflows/submodule-codebuild-ci.yml

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,12 @@ name: Sagemaker PR Checks
22
on:
33
pull_request_target:
44
branches:
5-
- "master-v3"
5+
- "master*"
66
paths:
77
- 'sagemaker-train/**'
88
- 'sagemaker-serve/**'
99
- 'sagemaker-mlops/**'
10+
- 'sagemaker-core/**'
1011

1112
concurrency:
1213
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.head_ref }}
@@ -55,7 +56,7 @@ jobs:
5556
- uses: actions/checkout@v3
5657
with:
5758
fetch-depth: 0
58-
token: ${{ secrets.REPO_ACCESS_KEY }} # or use appropriate token
59+
token: ${{ secrets.GH_PAT }} # or use appropriate token
5960
ref: ${{ github.event.pull_request.base.ref }} # Target branch (master-v3)
6061
- name: Detect Changes
6162
id: check-changes
@@ -93,6 +94,13 @@ jobs:
9394
SUBMODULES=$(echo $SUBMODULES | sed 's/\]$/,"sagemaker-mlops"\]/')
9495
fi
9596
fi
97+
if echo "$CHANGES" | grep -q "^sagemaker-core/"; then
98+
if [ "$SUBMODULES" = '[]' ]; then
99+
SUBMODULES='["sagemaker-core"]'
100+
else
101+
SUBMODULES=$(echo $SUBMODULES | sed 's/\]$/,"sagemaker-core"\]/')
102+
fi
103+
fi
96104
echo "Final SUBMODULES: $SUBMODULES"
97105
echo "submodules=$SUBMODULES" >> $GITHUB_OUTPUT
98106
@@ -101,6 +109,7 @@ jobs:
101109
needs: [detect-changes]
102110
if: needs.detect-changes.outputs.submodules != '[]'
103111
strategy:
112+
fail-fast: false
104113
matrix:
105114
submodule: ${{ fromJson(needs.detect-changes.outputs.submodules) }}
106115
steps:
@@ -144,6 +153,7 @@ jobs:
144153
needs: [detect-changes]
145154
if: needs.detect-changes.outputs.submodules != '[]'
146155
strategy:
156+
fail-fast: false
147157
matrix:
148158
submodule: ${{ fromJson(needs.detect-changes.outputs.submodules) }}
149159
steps:
@@ -158,4 +168,4 @@ jobs:
158168
uses: aws-actions/aws-codebuild-run-build@v1
159169
with:
160170
project-name: ${{ github.event.repository.name }}-ci-${{ matrix.submodule }}-integ-tests
161-
source-version-override: 'refs/pull/${{ github.event.pull_request.number }}/head^{${{ github.event.pull_request.head.sha }}}'
171+
source-version-override: 'refs/pull/${{ github.event.pull_request.number }}/head^{${{ github.event.pull_request.head.sha }}}'

CHANGELOG.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,13 @@
11
# Changelog
22

3+
## v3.1.0 (2025-12-03)
4+
5+
### Features
6+
7+
* Fine-tuning SDK: SFT, RLVR, and RLAIF techniques with standardized parameter design
8+
* AIRegistry Integration: Added CRUD operations for datasets and evaluators
9+
* Enhanced Training Experience: Implemented MLFlow metrics tracking and deployment workflows
10+
311
## v3.0.1 (2025-11-19)
412

513
* Update project dependencies to include submodules: sagemaker-core, sagemaker-train, sagemaker-serve, sagemaker-mlops

README.rst

Lines changed: 47 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -409,4 +409,50 @@ SageMaker V2 Examples
409409
#. `SageMaker Autopilot <src/sagemaker/automl/README.rst>`__
410410
#. `Model Monitoring <https://sagemaker.readthedocs.io/en/stable/amazon_sagemaker_model_monitoring.html>`__
411411
#. `SageMaker Debugger <https://sagemaker.readthedocs.io/en/stable/amazon_sagemaker_debugger.html>`__
412-
#. `SageMaker Processing <https://sagemaker.readthedocs.io/en/stable/amazon_sagemaker_processing.html>`__
412+
#. `SageMaker Processing <https://sagemaker.readthedocs.io/en/stable/amazon_sagemaker_processing.html>`__
413+
414+
🚀 Model Fine-Tuning Support Now Available in V3
415+
-------------------------------------------------
416+
417+
We're excited to announce model fine-tuning capabilities in SageMaker Python SDK V3!
418+
419+
**What's New**
420+
421+
Four new trainer classes for fine-tuning foundation models:
422+
423+
* SFTTrainer - Supervised fine-tuning
424+
* DPOTrainer - Direct preference optimization
425+
* RLAIFTrainer - RL from AI feedback
426+
* RLVRTrainer - RL from verifiable rewards
427+
428+
**Quick Example**
429+
430+
.. code:: python
431+
432+
from sagemaker.train import SFTTrainer
433+
from sagemaker.train.common import TrainingType
434+
435+
trainer = SFTTrainer(
436+
model="meta-llama/Llama-2-7b-hf",
437+
training_type=TrainingType.LORA,
438+
model_package_group_name="my-models",
439+
training_dataset="s3://bucket/train.jsonl"
440+
)
441+
442+
training_job = trainer.train()
443+
444+
**Key Features**
445+
446+
* ✨ LoRA & full fine-tuning
447+
* 📊 MLflow integration with real-time metrics
448+
* 🚀 Deploy to SageMaker or Bedrock
449+
* 📈 Built-in evaluation (11 benchmarks)
450+
* ☁️ Serverless training
451+
452+
**Get Started**
453+
454+
.. code:: python
455+
456+
pip install sagemaker>=3.1.0
457+
458+
`📓 Example notebooks <https://github.com/aws/sagemaker-python-sdk/tree/master/v3-examples/model-customization-examples>`__

VERSION

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
3.0.1
1+
3.1.0

pyproject.toml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -33,10 +33,10 @@ classifiers = [
3333
"Programming Language :: Python :: 3.12",
3434
]
3535
dependencies = [
36-
"sagemaker-core>=2.0.0,<3.0.0",
37-
"sagemaker-train<2.0.0",
38-
"sagemaker-serve<2.0.0",
39-
"sagemaker-mlops<2.0.0",
36+
"sagemaker-core>=2.1.0,<3.0.0",
37+
"sagemaker-train>=1.1.0,<2.0.0",
38+
"sagemaker-serve>=1.1.0,<2.0.0",
39+
"sagemaker-mlops>=1.1.0,<2.0.0",
4040
]
4141

4242
[project.optional-dependencies]
@@ -59,4 +59,4 @@ addopts = ["-vv"]
5959
testpaths = ["tests"]
6060

6161
[tool.black]
62-
line-length = 100
62+
line-length = 100

sagemaker-core/VERSION

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,2 @@
1-
2.0.1
1+
2.1.0
2+

sagemaker-core/pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ authors = [
1212
readme = "README.rst"
1313
dependencies = [
1414
# Add your dependencies here (Include lower and upper bounds as applicable)
15-
"boto3>=1.35.75,<2.0.0",
15+
"boto3>=1.42.2,<2.0.0",
1616
"pydantic>=2.0.0,<3.0.0",
1717
"PyYAML>=6.0, <7.0",
1818
"jsonschema<5.0.0",

0 commit comments

Comments
 (0)