← All reportsPublic report

pytorch/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

103k stars Python View on GitHubprofiled 15d ago
Try GitZoid
400PRs this week
168Contributors
0Deps scanned
0Issues found
01 · Repo overview

How pytorch is put together

PyTorch is a large open-source deep learning framework whose primary interface is the Python `torch` package backed by a massive C++/CUDA core. The build is orchestrated by CMake (root CMakeLists.txt plus per-directory CMakeLists under aten/, c10/, torch/, caffe2/) driven through scikit-build-core as the PEP 517 backend declared in pyproject.toml, with Buck/Bazel .bzl files maintained in parallel for some targets. Code generation is central: torchgen emits ATen operator bindings, and third_party wrapper generators produce cpuinfo/XNNPACK shim sources. The repo also contains an Android Gradle binding, a large CI system (.ci/docker, Jenkins entrypoints) that builds Docker images per BUILD_ENVIRONMENT, and developer tooling under tools/. Data flow is: user-facing Python API -> dispatcher/ATen C++ kernels -> optional GPU backends, with AOTInductor/mobile binaries produced from the same tree.

Languages

PythonC++CMakeStarlark (Buck/Bazel .bzl)ShellGroovy/Gradle (Android)

Frameworks

scikit-build-core (PEP 517 build backend)CMakeBuck2/Bazel (BUCK.oss / defs.bzl)Gradle + Android SDK (android/)Jenkins CIDocker

Infrastructure

Docker images built per BUILD_ENVIRONMENT (.ci/docker)Jenkins test entrypointsGitHub Actions plumbing (.github excluded from sdist)Devcontainers (.devcontainer/Dockerfile)Android NDK/mobile builds

Major components

torch (Python package)

User-facing Python API, tensor types, docs strings, config introspection, and the weights-only unpickler for safe checkpoint loading.

ATen (aten/)

Core C++ tensor/operator library with its own CMakeLists.txt providing the mathematical and dispatch kernels.

c10

Minimal core C++ library (tensor storage, device, error handling) shared across PyTorch backends.

torchgen / codegen

Build-time Python code generator invoked by cmake/Codegen.cmake to emit ATen bindings and native function registrations.

functorch

Composable function transforms (vmap/grad) shipped as a Python package inside the wheel.

caffe2

Legacy Caffe2 integration layer retained in the build tree with its own CMakeLists and CI scripts.

.ci (CI infrastructure)

Docker image builds, Jenkins/Caffe2 test entrypoints, and the lumen_ci Python CLI tool driving containerized builds.

third_party

Vendored dependencies (cpuinfo, XNNPACK, flatbuffers, ittapi, nccl, etc.) plus generator scripts producing platform-conditional wrapper sources.

Over the past month, PyTorch development has been dominated by compiler work (Dynamo/Inductor), with steady improvements to GPU kernel generation, autotuning, and AOTInductor deployment, alongside a burst of fixes for attention (SDPA) and distributed training. Hardware support broadened, particularly for AMD ROCm GPUs and Intel XPU devices, while a large cleanup effort removed dead code and fixed typos across the codebase. Activity tapered sharply in the final two weeks, dropping from hundreds of commits to just a handful.

Week by week

2026-08-24A busy week of compiler, distributed-training, and hardware-support improvements, including new Intel XPU dispatching and faster fused kernels on AMD's latest GPUs.latest6 changes

Feature

Intel XPU support for linalg.polar

The polar decomposition math operation can now run on Intel GPU hardware.

Refactor

Faster matrix multiplication on AMD gfx950 GPUs

New tuned GEMM kernels were added to speed up torch.mm on AMD's latest accelerator architecture.

Feature

SimpleNamespace support in Dynamo

torch.compile can now trace Python's types.SimpleNamespace objects without breaking.

Fix

Fixed pdist/cdist failures on large ROCm outputs

Distance computations that produce more than ~16 million results no longer crash on AMD GPUs.

Refactor

Fused epilogues in nested reductions

Inductor now merges extra computation steps into reduction loops to reduce memory traffic and speed up models.

Chore

Removed the deprecated assert_allclose helper

A long-deprecated testing utility was removed, which is a breaking change for anyone still calling it.

2026-08-17A very high-volume week centered on Dynamo reliability fixes, enabling torch.compile for RNNs by default, and new stateless random-number APIs.6 changes

Feature

torch.compile enabled for RNN modules by default

Recurrent neural network layers are now automatically accelerated by the compiler instead of falling back to eager execution.

Feature

Stateless RNG APIs for randint and raw bits

New deterministic random-number generators let users reproduce exact random sequences independent of global state.

Fix

Fixed efficient attention when queries exceed keys

Bottom-right aligned attention no longer produces wrong results when the query sequence is longer than the key sequence.

Fix

Fixed scan operation autograd

Gradients through the scan control-flow primitive now compute correctly.

Refactor

Vendor QuACK symmetric GEMM into Inductor

An optimized symmetric matrix-multiplication kernel was integrated to speed up relevant workloads.

Refactor

Sweep of typo and grammar cleanups

Roughly ten separate commits cleaned up comments and docstrings across the codebase for readability.

2026-08-10Mostly maintenance and cleanup: dead code removal from Apple MPS backend, test refactoring for device-agnostic CI, plus several correctness fixes.6 changes

Refactor

Dead code removal from MPS backend (5 parts)

Five sequential commits stripped out unused code from the Apple Silicon GPU backend, shrinking its maintenance surface.

Fix

Fixed 64-bit indexing in spatial softmax backward

Softmax gradient computation no longer breaks on very large tensors that exceed 32-bit index limits.

Fix

Fixed FFT operations with empty batch dimensions

Fourier transforms now handle inputs with zero-sized batch dimensions gracefully instead of erroring.

Feature

Deterministic upsample bicubic backward

Bicubic upsampling gradients can now be computed deterministically for reproducible training runs.

Refactor

Speeded up B200 tests with pytest-xdist

Continuous-integration tests on NVIDIA B200 hardware now run in parallel, cutting wait times.

Fix

Preserved storage sharing when loading checkpoints onto fake devices

Loading saved models onto meta/fake devices keeps tensors sharing memory as intended, avoiding subtle bugs.

2026-08-03A very quiet week with only two small Inductor fixes shipped.2 changes

Fix

Fixed EXHAUSTIVE search space check in tuned_mm

Corrected how the matrix-multiply autotuner validates its most thorough search mode.

Fix

Prevented recursive handle_wait in overlap scheduling

Stopped an infinite recursion bug in Inductor's communication/computation overlap logic.

2026-07-27A single commit landed this week, fixing how autograd saves mutated gradient inputs during compilation.1 change

Fix

Stopped saving mutated grad inputs in AOT min-cut

Ahead-of-time compilation no longer needlessly retains modified gradient inputs, reducing memory overhead in compiled training graphs.

03 · Security check

Dependencies and code review

0 dependencies scanned

Dependency advisories

Security Watch

No known advisories across 0 scanned dependencies.

No known advisories in the scanned dependencies.

Code review

No concrete code-level issues confirmed.

Get this report every week for your repos.

GitZoid learns each repo, reports what changed, and flags what needs attention. One flat price for the whole team.

$19 a month, flat · First 10 outputs free · No card required