PyTorch 2.1: automatic dynamic shape compilation, distributed checkpointing

Imported from official source

Software release

AI Classified by Officially

# PyTorch 2.1 Release Notes - Highlights - Backwards Incompatible Change - Deprecations - New Features - Improvements - Bug fixes - Performance - Documentation - Developers - Security # Highlights We are excited to announce the release of PyTorch® 2.1! PyTorch 2.1 offers automatic dynamic shape support in torch.compile, torch.distributed.checkpoint for saving/loading distributed training jobs on multiple ranks in parallel, and torch.compile support for the NumPy API. In addition, this release offers numerous performance improvements (e.g. CPU inductor improvements, AVX512 support, scaled-dot-product-attention support) as well as a prototype release of torch.export, a sound full-graph capture mechanism, and `torch.export`-based quantization. Along with 2.1, we are also releasing a series of updates to the PyTorch domain libraries. More details can be found in the library updates blog. This release is composed of 6,682 commits and 784 contributors since 2.0. We want to sincerely thank our dedicated community for your contributions. As always, we encourage you to try these out and report any issues as we improve 2.1. More information about how to get started with the PyTorch 2-series can be found at our [Getting Started](https://pytorch.org/get-started/pytorch-2.0/) page. Summary: - `torch.compile` now includes automatic support for detecting and minimizing recompilations due to tensor shape changes using automatic dynamic shapes. - `torch.distributed.checkpoint` enables saving and loading models from multiple ranks in parallel, as well as resharding due to changes in cluster topology. - `torch.compile` can now compile NumPy operations via translating them into PyTorch-equivalent operations. - `torch.compile` now includes improved support for Python 3.11. - New CPU performance features include inductor improvements (e.g. bfloat16 support and dynamic shapes), AVX512 kernel support, and scaled-dot-product-attention kernels.

This is an extract. The publication continues at the source.

Read the original at the source: https://github.com/pytorch/pytorch/releases/tag/v2.1.0 This address stopped responding when last checked on October 03, 2026. The record below is what Officially captured.

Officially imported this from PyTorch’s own source and shows an extract. If you work there, claiming the profile and verifying the domain lets you choose to show the full text here.

This publication has changed since it was first published

2 versions recorded. The original is kept in full — nothing is overwritten.

  1. v2 imported change on current
  2. v1 as first published on

Provenance

Organization
PyTorch — imported from official source
Official source
https://api.github.com/repos/pytorch/pytorch/releases?per_page=25 API
Imported
September 15, 2026 19:08
Versions
2 recorded
Identity
123579416

Officially records where a publication came from, not whether it is true. Imported records are reproduced from an organization's own official source.