release: v3.2.1

[github-actions]

This pull request was created by GitHub Actions/AWS CodeBuild! Before merging, please do the following:

• Review changelog/staleness report.
• Review pinned_env.in files, when releasing Major/Minor version, try removing entries from there and build images.
• Review python package dependency and size report.
• Review build/test results by clicking Build Logs in CI Report (be patient, tests take ~4hr).
• Review ECR Scan results.

[sagemaker-distribution-bot]

AWS CodeBuild CI Report

• CodeBuild project: CosmosImageBuildersagemaker-U5omrjiHQCoa
• Commit ID: bbabbc7
• Result: FAILED
• Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

[sagemaker-distribution-bot]

Staleness Report: 3.2.1(gpu)

Package	Current Version in the Distribution image	Latest Relevant Version in Upstream
numpy	1.26.4	1.26.4
jinja2	3.1.6	3.1.6
pytorch	2.6.0	2.6.0
pandas	2.2.3	2.2.3
${\color{red}boto3}$	1.37.1	1.37.38
aioboto3	14.1.0	14.1.0
altair	5.5.0	5.5.0
ipython	8.36.0	8.36.0
jupyter-lsp	2.2.5	2.2.5
jupyterlab	4.4.3	4.4.3
amazon-q-developer-jupyterlab-ext	3.4.8	3.4.8
langchain	0.3.25	0.3.25
jupyter-ai	2.31.5	2.31.5
amazon-sagemaker-jupyter-ai-q-developer	1.2.7	1.2.7
jupyter-scheduler	2.11.0	2.11.0
amazon-sagemaker-jupyter-scheduler	3.1.14	3.1.14
amazon-sagemaker-sql-magic	0.1.4	0.1.4
jupyterlab-lsp	5.0.3	5.0.3
amazon_sagemaker_sql_editor	0.1.18	0.1.18
amzn-sagemaker-aiops-jupyterlab-extension	1.0.4	1.0.4
scipy	1.15.2	1.15.2
matplotlib-base	3.10.3	3.10.3
scikit-learn	1.6.1	1.6.1
pip	25.1.1	25.1.1
torchvision	0.21.0	0.21.0
autogluon	1.3.1	1.3.1
ipywidgets	8.1.7	8.1.7
notebook	7.4.3	7.4.3
aws-glue-sessions	1.0.9	1.0.9
conda	25.5.1	25.5.1
docker-cli	27.5.1	27.5.1
uvicorn	0.34.3	0.34.3
fastapi	0.115.12	0.115.12
git-remote-codecommit	1.16	1.16
jupyter-activity-monitor-extension	0.3.1	0.3.1
jupyter-collaboration	3.1.0	3.1.0
jupyter-dash	0.4.2	0.4.2
jupyter-server-proxy	4.4.0	4.4.0
jupyterlab-git	0.51.2	0.51.2
keras	3.10.0	3.10.0
${\color{red}langchain-aws}$	0.2.19	0.2.25
mlflow	2.22.0	2.22.0
py-xgboost-gpu	2.1.4	2.1.4
pyhive	0.7.0	0.7.0
python-gssapi	1.9.0	1.9.0
python-lsp-server	1.12.2	1.12.2
s3fs	2024.12.0	2024.12.0
sagemaker-code-editor	1.5.0	1.5.0
sagemaker-headless-execution-driver	0.0.13	0.0.13
sagemaker-jupyterlab-emr-extension	0.3.10	0.3.10
sagemaker-jupyterlab-extension	0.4.2	0.4.2
sagemaker-kernel-wrapper	0.0.5	0.0.5
sagemaker-mlflow	0.1.0	0.1.0
sagemaker-python-sdk	2.245.0	2.245.0
sagemaker-studio	1.0.15	1.0.15
sagemaker-studio-analytics-extension	0.1.7	0.1.7
sagemaker-studio-cli	1.0.6	1.0.6
sagemaker-studio-dataengineering-extensions	1.0.11	1.0.11
sagemaker-studio-dataengineering-sessions	1.0.13	1.0.13
seaborn	0.13.2	0.13.2
supervisor	4.2.5	4.2.5
tensorflow	2.18.0	2.18.0
tf-keras	2.18.0	2.18.0

Staleness Report: 3.2.1(cpu)

Package	Current Version in the Distribution image	Latest Relevant Version in Upstream
numpy	1.26.4	1.26.4
jinja2	3.1.6	3.1.6
pytorch	2.6.0	2.6.0
pandas	2.2.3	2.2.3
${\color{red}boto3}$	1.37.1	1.37.38
aioboto3	14.1.0	14.1.0
altair	5.5.0	5.5.0
ipython	8.36.0	8.36.0
jupyter-lsp	2.2.5	2.2.5
jupyterlab	4.4.3	4.4.3
amazon-q-developer-jupyterlab-ext	3.4.8	3.4.8
langchain	0.3.25	0.3.25
jupyter-ai	2.31.5	2.31.5
amazon-sagemaker-jupyter-ai-q-developer	1.2.7	1.2.7
jupyter-scheduler	2.11.0	2.11.0
amazon-sagemaker-jupyter-scheduler	3.1.14	3.1.14
amazon-sagemaker-sql-magic	0.1.4	0.1.4
jupyterlab-lsp	5.0.3	5.0.3
amazon_sagemaker_sql_editor	0.1.18	0.1.18
amzn-sagemaker-aiops-jupyterlab-extension	1.0.4	1.0.4
scipy	1.15.2	1.15.2
matplotlib-base	3.10.3	3.10.3
scikit-learn	1.6.1	1.6.1
pip	25.1.1	25.1.1
torchvision	0.21.0	0.21.0
autogluon	1.3.1	1.3.1
ipywidgets	8.1.7	8.1.7
notebook	7.4.3	7.4.3
aws-glue-sessions	1.0.9	1.0.9
conda	25.5.1	25.5.1
docker-cli	27.5.1	27.5.1
uvicorn	0.34.3	0.34.3
fastapi	0.115.12	0.115.12
git-remote-codecommit	1.16	1.16
jupyter-activity-monitor-extension	0.3.1	0.3.1
jupyter-collaboration	3.1.0	3.1.0
jupyter-dash	0.4.2	0.4.2
jupyter-server-proxy	4.4.0	4.4.0
jupyterlab-git	0.51.2	0.51.2
keras	3.10.0	3.10.0
${\color{red}langchain-aws}$	0.2.19	0.2.25
mlflow	2.22.0	2.22.0
py-xgboost-cpu	2.1.4	2.1.4
pyhive	0.7.0	0.7.0
python-gssapi	1.9.0	1.9.0
python-lsp-server	1.12.2	1.12.2
s3fs	2024.12.0	2024.12.0
sagemaker-code-editor	1.5.0	1.5.0
sagemaker-headless-execution-driver	0.0.13	0.0.13
sagemaker-jupyterlab-emr-extension	0.3.10	0.3.10
sagemaker-jupyterlab-extension	0.4.2	0.4.2
sagemaker-kernel-wrapper	0.0.5	0.0.5
sagemaker-mlflow	0.1.0	0.1.0
sagemaker-python-sdk	2.245.0	2.245.0
sagemaker-studio	1.0.15	1.0.15
sagemaker-studio-analytics-extension	0.1.7	0.1.7
sagemaker-studio-cli	1.0.6	1.0.6
sagemaker-studio-dataengineering-extensions	1.0.11	1.0.11
sagemaker-studio-dataengineering-sessions	1.0.13	1.0.13
seaborn	0.13.2	0.13.2
supervisor	4.2.5	4.2.5
tensorflow	2.18.0	2.18.0
tf-keras	2.18.0	2.18.0

[sagemaker-distribution-bot]

GPU Total Image Size Report

Target Image Size	Base Image Size	Size Difference	Size Change (%)
3.2.1-gpu:7.06 GB	3.2.0-gpu:7.03 GB	0.03 GB Larger	0.37% Larger

CPU Total Image Size Report

Target Image Size	Base Image Size	Size Difference	Size Change (%)
3.2.1-cpu:2.12 GB	3.2.0-cpu:2.13 GB	-0.01 GB Smaller	0.42% Smaller

Python Package Size Report (GPU)

Target Image Version: 3.2.1 | Base Image Version: 3.2.0

Python Packages Total Size Summary

Target Version Total Size	Base Version Total Size	Size Change (abs)	Size Change (%)
4.25GB	4.22GB	33.63MB	0.78

Top-20 Largest Python Packages

Package	Version in the Target Image	Size
libcudnn	9.10.1.4	502.61MB
libtorch	2.6.0	498.48MB
tensorflow-base	2.18.0	364.24MB
libmagma	2.8.0	282.34MB
libcublas	12.6.4.1	256.23MB
nccl	2.27.3.1	203.74MB
libcufft	11.3.0.4	156.19MB
onnxruntime	1.22.0	134.30MB
libxgboost	2.1.4	128.62MB
mkl	2024.2.2	118.94MB
libcusparse	12.5.4.2	118.64MB
triton	3.2.0	97.99MB
libcusolver	11.7.1.2	95.84MB
libnpp	12.3.1.54	93.37MB
amazon-sagemaker-jupyter-ai-q-developer	1.2.7	75.06MB
gcc_impl_linux-64	13.3.0	63.68MB
cuda-nvdisasm	12.6.77	47.62MB
catboost	1.2.8	42.56MB
sagemaker-code-editor	1.5.0	41.48MB
libllvm20	20.1.7	41.03MB

Python Package Size Delta

The total size of newly introduced Python packages is 1009.46KB, accounts for 0.02% of the total package size.

Package	Version in the Target Image	Version in the Base Image	Size Change (abs)	Size Change (%)
nccl	2.27.3.1	2.26.6.1	31.60MB	18.36
llvm-openmp	20.1.7	20.1.6	942.57KB	30.16
beartype	0.21.0	-	930.61KB	-
libmamba	2.3.0	2.1.1	318.94KB	15.88
networkx	3.5	3.4.2	292.44KB	23.67
aiohttp	3.12.13	3.9.5	194.76KB	24.82
libmambapy	2.3.0	2.1.1	104.36KB	17.08
plum-dispatch	2.5.7	-	39.47KB	-
fasttransform	0.0.2	-	20.10KB	-
aiohappyeyeballs	2.6.1	-	19.29KB	-
libarchive	3.8.1	3.7.7	16.63KB	1.97
pytest	8.4.0	8.3.5	14.84KB	5.85
graphviz	13.0.1	12.2.1	7.58KB	0.32
boltons	25.0.0	24.0.0	4.72KB	1.63
libllvm20	20.1.7	20.1.6	3.65KB	0.01
pixman	0.46.2	0.46.0	3.47KB	0.89
databricks-sdk	0.56.0	0.55.0	2.84KB	0.66
narwhals	1.43.0	1.41.0	2.59KB	1.18
c-blosc2	2.18.0	2.17.1	2.59KB	0.77
conda	25.5.1	25.5.0	2.28KB	0.2
jupyter-ai-magics	2.31.5	2.31.4	1.95KB	5.52
fonttools	4.58.4	4.58.1	1.72KB	0.06
libcurl	8.14.1	8.14.0	1.62KB	0.37
graphite2	1.3.14	1.3.13	1.53KB	1.61
tomlkit	0.13.3	0.13.2	1.37KB	3.76
huggingface_hub	0.33.0	0.32.3	1.36KB	0.44
libaec	1.1.4	1.1.3	1.35KB	3.89
opentelemetry-semantic-conventions	0.55b1	0.54b1	1.27KB	1.29
opencl-headers	2025.06.13	2024.10.24	1.27KB	2.4
fastai	2.8.2	2.7.19	1.22KB	0.67
greenlet	3.2.3	3.2.2	1.08KB	0.47
nltk	3.9.1	3.8.2	1.07KB	0.1
pydantic	2.11.7	2.11.4	1.00KB	0.34
jupyter-ai	2.31.5	2.31.4	976.00B	0.4
sagemaker-studio	1.0.15	1.0.14	904.00B	0.55
libpng	1.6.49	1.6.47	805.00B	0.28
libsystemd0	257.6	257.4	791.00B	0.16
requests	2.32.4	2.32.3	684.00B	1.16
pytest-cov	6.2.1	6.1.1	651.00B	2.36
google-api-core	2.25.1	2.24.2	617.00B	0.68
python-graphviz	0.21	0.20.3	617.00B	1.61
amazon-sagemaker-sql-execution	0.1.8	0.1.7	530.00B	1.53
libudev1	257.6	257.4	492.00B	0.34
typing-extensions	4.14.0	4.13.2	410.00B	0.46
pycrdt	0.12.21	0.12.19	372.00B	0.05
param	2.2.1	2.2.0	366.00B	0.35
sagemaker-jupyterlab-emr-extension	0.3.10	0.3.8	327.00B	0.11
tifffile	2025.6.11	2025.5.26	274.00B	0.15
zipp	3.23.0	3.22.0	272.00B	1.2
jupyterlab-git	0.51.2	0.51.1	177.00B	0.06
google-auth	2.40.3	2.40.2	168.00B	0.14
setuptools	80.9.0	80.8.0	167.00B	0.02
simdjson	3.13.0	3.12.3	143.00B	0.06
argon2-cffi	25.1.0	23.1.0	121.00B	0.65
uvicorn	0.34.3	0.34.2	57.00B	0.12
python-gil	3.12.11	3.12.10	5.00B	0.01
cpython	3.12.11	3.12.10	-9.00B	-0.02
uvicorn-standard	0.34.3	0.34.2	-48.00B	-0.63
sagemaker-jupyterlab-extension-common	0.2.5	0.2.4	-61.00B	-0.03
amazon-sagemaker-jupyter-scheduler	3.1.14	3.1.11	-65.00B	-0.04
amzn-sagemaker-aiops-jupyterlab-extension	1.0.4	1.0.3	-136.00B	-0.44
opentelemetry-api	1.34.1	1.33.1	-164.00B	-0.36
xkeyboard-config	2.45	2.44	-464.00B	-0.12
torchmetrics	1.7.3	1.7.2	-514.00B	-0.13
ruamel.yaml	0.18.14	0.18.11	-549.00B	-0.2
libsolv	0.7.33	0.7.32	-551.00B	-0.12
coverage	7.9.1	7.8.2	-615.00B	-0.17
prometheus_client	0.22.1	0.22.0	-873.00B	-1.63
pyzmq	27.0.0	26.4.0	-944.00B	-0.25
typing_extensions	4.14.0	4.13.2	-1.18KB	-2.32
ca-certificates	2025.6.15	2025.4.26	-1.19KB	-0.8
opentelemetry-sdk	1.34.1	1.33.1	-1.26KB	-1.63
cryptography	45.0.4	45.0.3	-1.33KB	-0.08
watchfiles	1.1.0	1.0.5	-1.61KB	-0.39
certifi	2025.6.15	2025.4.26	-1.78KB	-1.16
sagemaker-core	1.0.37	1.0.35	-2.13KB	-0.83
msgpack-python	1.1.1	1.1.0	-2.29KB	-2.23
fastcore	1.8.2	1.7.20	-4.01KB	-4.9
yarl	1.20.1	1.20.0	-6.00KB	-3.95
imagecodecs	2025.3.30	2024.12.30	-8.14KB	-0.43
amazon_sagemaker_sql_editor	0.1.18	0.1.16	-23.61KB	-0.13
h5py	3.14.0	3.13.0	-68.08KB	-5.02
hf-xet	1.1.3	1.1.2	-298.84KB	-13.78
amazon-sagemaker-jupyter-ai-q-developer	1.2.7	1.2.4	-611.16KB	-0.79

Python Package Size Report (CPU)

Target Image Version: 3.2.1 | Base Image Version: 3.2.0

Python Packages Total Size Summary

Target Version Total Size	Base Version Total Size	Size Change (abs)	Size Change (%)
1.14GB	1.14GB	1.82MB	0.16

Top-20 Largest Python Packages

Package	Version in the Target Image	Size
tensorflow-base	2.18.0	150.67MB
mkl	2024.2.2	118.94MB
amazon-sagemaker-jupyter-ai-q-developer	1.2.7	75.06MB
libtorch	2.6.0	51.85MB
sagemaker-code-editor	1.5.0	41.48MB
pillow	11.2.1	40.54MB
python	3.12.9	30.12MB
ray-core	2.44.1	29.98MB
llvmlite	0.44.0	28.61MB
pytorch	2.6.0	27.02MB
panel	1.7.1	21.55MB
pandoc	3.7.0.2	20.70MB
amazon_sagemaker_sql_editor	0.1.18	17.11MB
nodejs	20.18.1	16.36MB
scipy	1.15.2	16.27MB
pandas	2.2.3	14.68MB
onnxruntime	1.22.0	12.39MB
icu	75.1	11.57MB
statsmodels	0.14.4	11.54MB
scikit-image	0.25.2	10.28MB

Python Package Size Delta

The total size of newly introduced Python packages is 1009.46KB, accounts for 0.08% of the total package size.

Package	Version in the Target Image	Version in the Base Image	Size Change (abs)	Size Change (%)
llvm-openmp	20.1.7	20.1.6	942.57KB	30.16
beartype	0.21.0	-	930.61KB	-
libmamba	2.3.0	2.1.1	318.94KB	15.88
networkx	3.5	3.4.2	292.44KB	23.67
aiohttp	3.12.13	3.9.5	194.76KB	24.82
libmambapy	2.3.0	2.1.1	104.36KB	17.08
plum-dispatch	2.5.7	-	39.47KB	-
fasttransform	0.0.2	-	20.10KB	-
aiohappyeyeballs	2.6.1	-	19.29KB	-
libarchive	3.8.1	3.7.7	16.63KB	1.97
pytest	8.4.0	8.3.5	14.84KB	5.85
graphviz	13.0.1	12.2.1	7.58KB	0.32
boltons	25.0.0	24.0.0	4.72KB	1.63
pixman	0.46.2	0.46.0	3.47KB	0.89
databricks-sdk	0.56.0	0.55.0	2.84KB	0.66
narwhals	1.43.0	1.41.0	2.59KB	1.18
c-blosc2	2.18.0	2.17.1	2.59KB	0.77
conda	25.5.1	25.5.0	2.28KB	0.2
jupyter-ai-magics	2.31.5	2.31.4	1.95KB	5.52
fonttools	4.58.4	4.58.1	1.72KB	0.06
libcurl	8.14.1	8.14.0	1.62KB	0.37
graphite2	1.3.14	1.3.13	1.53KB	1.61
tomlkit	0.13.3	0.13.2	1.37KB	3.76
huggingface_hub	0.33.0	0.32.3	1.36KB	0.44
libaec	1.1.4	1.1.3	1.35KB	3.89
opentelemetry-semantic-conventions	0.55b1	0.54b1	1.27KB	1.29
opencl-headers	2025.06.13	2024.10.24	1.27KB	2.4
fastai	2.8.2	2.7.19	1.22KB	0.67
greenlet	3.2.3	3.2.2	1.08KB	0.47
nltk	3.9.1	3.8.2	1.07KB	0.1
pydantic	2.11.7	2.11.4	1.00KB	0.34
jupyter-ai	2.31.5	2.31.4	976.00B	0.4
sagemaker-studio	1.0.15	1.0.14	904.00B	0.55
libpng	1.6.49	1.6.47	805.00B	0.28
requests	2.32.4	2.32.3	684.00B	1.16
pytest-cov	6.2.1	6.1.1	651.00B	2.36
google-api-core	2.25.1	2.24.2	617.00B	0.68
python-graphviz	0.21	0.20.3	617.00B	1.61
amazon-sagemaker-sql-execution	0.1.8	0.1.7	530.00B	1.53
typing-extensions	4.14.0	4.13.2	410.00B	0.46
ruamel.yaml	0.18.14	0.18.12	392.00B	0.15
pycrdt	0.12.21	0.12.19	372.00B	0.05
param	2.2.1	2.2.0	366.00B	0.35
sagemaker-jupyterlab-emr-extension	0.3.10	0.3.8	327.00B	0.11
tifffile	2025.6.11	2025.5.26	274.00B	0.15
zipp	3.23.0	3.22.0	272.00B	1.2
jupyterlab-git	0.51.2	0.51.1	177.00B	0.06
google-auth	2.40.3	2.40.2	168.00B	0.14
setuptools	80.9.0	80.8.0	167.00B	0.02
simdjson	3.13.0	3.12.3	143.00B	0.06
argon2-cffi	25.1.0	23.1.0	121.00B	0.65
uvicorn	0.34.3	0.34.2	57.00B	0.12
python-gil	3.12.11	3.12.10	5.00B	0.01
cpython	3.12.11	3.12.10	-9.00B	-0.02
uvicorn-standard	0.34.3	0.34.2	-48.00B	-0.63
sagemaker-jupyterlab-extension-common	0.2.5	0.2.4	-61.00B	-0.03
amazon-sagemaker-jupyter-scheduler	3.1.14	3.1.11	-65.00B	-0.04
amzn-sagemaker-aiops-jupyterlab-extension	1.0.4	1.0.3	-136.00B	-0.44
opentelemetry-api	1.34.1	1.33.1	-164.00B	-0.36
xkeyboard-config	2.45	2.44	-464.00B	-0.12
torchmetrics	1.7.3	1.7.2	-514.00B	-0.13
libsolv	0.7.33	0.7.32	-551.00B	-0.12
coverage	7.9.1	7.8.2	-615.00B	-0.17
prometheus_client	0.22.1	0.22.0	-873.00B	-1.63
pyzmq	27.0.0	26.4.0	-944.00B	-0.25
typing_extensions	4.14.0	4.13.2	-1.18KB	-2.32
ca-certificates	2025.6.15	2025.4.26	-1.19KB	-0.8
opentelemetry-sdk	1.34.1	1.33.1	-1.26KB	-1.63
cryptography	45.0.4	45.0.3	-1.33KB	-0.08
watchfiles	1.1.0	1.0.5	-1.61KB	-0.39
certifi	2025.6.15	2025.4.26	-1.78KB	-1.16
sagemaker-core	1.0.37	1.0.35	-2.13KB	-0.83
msgpack-python	1.1.1	1.1.0	-2.29KB	-2.23
fastcore	1.8.2	1.7.20	-4.01KB	-4.9
yarl	1.20.1	1.20.0	-6.00KB	-3.95
imagecodecs	2025.3.30	2024.12.30	-8.14KB	-0.43
amazon_sagemaker_sql_editor	0.1.18	0.1.16	-23.61KB	-0.13
h5py	3.14.0	3.13.0	-68.08KB	-5.02
hf-xet	1.1.3	1.1.2	-298.84KB	-13.78
amazon-sagemaker-jupyter-ai-q-developer	1.2.7	1.2.4	-611.16KB	-0.79

[sagemaker-distribution-bot]

Skipping dependency report for patch release.

[sagemaker-distribution-bot]

AWS CodeBuild CI Report

• CodeBuild project: CosmosImageBuildersagemaker-U5omrjiHQCoa
• Commit ID: 42054a6
• Result: FAILED
• Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

[sagemaker-distribution-bot]

AWS CodeBuild CI Report

• CodeBuild project: CosmosImageBuildersagemaker-U5omrjiHQCoa
• Commit ID: a32df85
• Result: FAILED
• Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

[sagemaker-distribution-bot]

AWS CodeBuild CI Report

• CodeBuild project: CosmosImageBuildersagemaker-U5omrjiHQCoa
• Commit ID: 61fed05
• Result: FAILED
• Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository