TensorFlow 2.21 Migration - SageMaker#6107

Open

bhanutejagk wants to merge 16 commits into

mainfrom

tensorflow-2.21-currency

Commits on May 18, 2026

TensorFlow 2.21 Migration - SageMaker

Bhanu Teja Goshikonda
and
claude
committed
Fix pre-commit hook failures

Bhanu Teja Goshikonda
and
claude
committed
Strip trailing newline from Dockerfiles per dockerfmt

Bhanu Teja Goshikonda
and
claude
committed
Fix telemetry framework value + populate CVE allowlist

Bhanu Teja Goshikonda
and
claude
committed

Commits on May 19, 2026

Add sub-phase B test scaffolding (unit + SageMaker integration)

Bhanu Teja Goshikonda
and
claude
committed
Fix unit-test failures from first run, mirror PT patterns

Bhanu Teja Goshikonda
and
claude
committed
Fix CUDA unit-test symlink chain + add SM diagnostics

Bhanu Teja Goshikonda
and
claude
committed
Generate SSH host keys so sshd can start in SM training

Bhanu Teja Goshikonda
and
claude
committed
Refresh CVE allowlist: 32 patched, 1 new

Bhanu Teja Goshikonda
and
claude
committed
Restore CVE allowlist after premature trim

Bhanu Teja Goshikonda
and
claude
committed
Fix TF_CONFIG cluster spec for multi-GPU multi-node training

Bhanu Teja Goshikonda
and
claude
committed

Commits on May 20, 2026

Drop MPI launcher; use TF_CONFIG + MultiWorkerMirroredStrategy directly

Bhanu Teja Goshikonda
and
claude
committed
Distribute MNIST dataset via strategy.distribute_datasets_from_function

Bhanu Teja Goshikonda
and
claude
committed
Pass plain dataset to model.fit; use AutoShardPolicy.DATA

Bhanu Teja Goshikonda
and
claude
committed
Restructure SM tests to mirror master coverage (SDK v3 translation)

Bhanu Teja Goshikonda
and
claude
committed
Bump CUDA base image from 12.6.3 to 12.9.1 (TF 2.21 SageMaker GPU)

Bhanu Teja Goshikonda
and
claude
committed