Tool Solutions: ML Frameworks r26.02

PyTorch build

• The pytorch-aarch64 r26.02 README.md is available here.
• A Docker image is available on Docker Hub under armlimited/pytorch-arm-neoverse.
• Attached as artefacts are:
  • A full Software Bill of Materials (SBOM) as Tool-Solutions_r26.02_SPDX_SBOM.json, and;
  • License texts as Tool-Solutions_r26.02_licenses.zip which cover all packages included in the built image, in addition to the base Ubuntu image.

Added

• Adds PyTorch PR #170600, to patch incremental build support.
• Adds PyTorch PR #170062, to add ccache support to ACL/OpenBLAS and manywheel build script.

Changed

• Updates hashes for:
  • PYTORCH_HASH to 77da53a7356e033e3fc1e03fdd960fc4ad117882, 2.11.0.dev20260129 from viable/strict, Jan 29th.
  • IDEEP_HASH to bbb9ffb9e0c401ca058b7f35a6ebe7d0e08ffd34, from ideep_pytorch, Jan 30th.
  • ONEDNN_HASH to 804f364c04ad8a763d534abaabc99bf99c2754e0, from main, Jan 30th.
  • TORCH_AO_HASH to 30fcb156945ecacd515775414d37c09bfe60727e, from main, Jan 30th.
  • KLEIDIAI_HASH to 5addaad73ebbb02e7dde6c50fff3bdb2ae8c407f, v1.20.0 from main, Jan 30th.
• Replaces ACL_VERSION=v52.6.0 with ACL_VERSION=v52.8.0, from main, Jan 30th.
• Updates torchvision from 0.25.0.dev20251104 to 0.25.0.dev20260129.
• Updates PyTorch and TorchAO manylinux AArch64 builder image from cpu-aarch64-d8be0384e085f551506bd739678109fa0f5ee7ac
  to cpu-aarch64-69d4c1f80b5e7da224d4f9c2170ef100e75dfe03.
• Updates the URL used to download weights for ResNet50 used by classify_image.py.
• Updates Ubuntu and Python version to 24.04 and 3.12, respectively.

Removed

• Removes patches which are no longer required or have landed upstream.

TensorFlow build

Changed

• Updates TensorFlow hash to 3d4958d107b77a2e4398c066971a099a369ef739, from nightly, Feb 2nd
• Updates Ubuntu and Python version to 24.04 and 3.12, respectively.

Note

Running tensorflow-aarch64/examples/run_unit_tests.sh with ONEDNN_DEFAULT_FPMATH_MODE=BF16 can lead to some failing tests. Addressing these failures is a WIP. However, all testing is expected to pass with the default setting ONEDNN_DEFAULT_FPMATH_MODE=FP32.

What's Changed

• Bump torchao builder image to match torch by @puneetmatharu in #414
• Add woke support by @puneetmatharu in #416
• Add ccache support by @puneetmatharu in #415
• Bump Ubuntu (24.04) and Python (3.12) versions by @puneetmatharu in #409
• Bump nightly torchvision version to 0.25.0.dev20260111 by @puneetmatharu in #417
• Bump TensorFlow commit hash by @puneetmatharu in #418
• Commit files changed/added in apply-github-patch by @puneetmatharu in #382
• Switch to slim Python Docker image by @puneetmatharu in #419
• Update ResNet50 download link by @puneetmatharu in #420
• Make project REUSE compliant + add pre-commit hooks by @puneetmatharu in #413
• Patch torchao build by @puneetmatharu in #421
• Bump builder images to get newer libgomp by @puneetmatharu in #422
• Merge main into beta-test-debian-pytorch-image by @puneetmatharu in #424
• Bump commit hashes for r26.02 and list .ccache/ in .gitignore by @danwhittaker-arm in #423
• Pull in bumped r26.02 hashes to beta testing branch by @puneetmatharu in #427
• Tidy up CHANGELOGs and add a trailing newline by @puneetmatharu in #426
• Tidy up pytorch-aarch64 build scripts by @puneetmatharu in #425
• Switch to slim Python/Debian image for PyTorch runner env by @puneetmatharu in #428
• Remove mslk from pytorch/third_party/ by @puneetmatharu in #429
• Add dummy credentials to apply-github-patch usage by @puneetmatharu in #431
• Updates CHANGELOGs for 26.02 release by @danwhittaker-arm in #432
• Fix uv version to 0.9.29 by @puneetmatharu in #433
• Patch build for internal CI by @puneetmatharu in #435
• Remove USERNAME arg from PyTorch Docker build by @puneetmatharu in #436
• Refactor dep versions into versions.sh and patch torch wheel naming by @puneetmatharu in #439

New Contributors

• @danwhittaker-arm made their first contribution in #423

Full Changelog: r25.12...r26.02