HighVulnerability

CVE-2026-31214: torch-checkpoint-shrink.py insecure deserialization of PyTorch checkpoint files

Source: NVD(opens in a new tab)
Identifier: CVE-2026-31214
Published: May 12, 2026
Record updated: October 9, 2026

Known exploitation: Not listed in the CISA Known Exploited Vulnerabilities catalog at the last check.
EPSS: 0.5%

Summary

The torch-checkpoint-shrink.py script in the ml-engineering project, at commit 0099885db36a8f06556efe1faf552518852cb1e0, is affected by CVE-2026-31214 (CWE-502). It calls torch.load() on PyTorch checkpoint (.pt) files without setting weights_only=True, so the pickle module can deserialize arbitrary Python objects. A remote attacker who supplies a maliciously crafted checkpoint file can achieve arbitrary code execution as the user running the script.

Mitigation

The source does not state a fix yet. Check the original advisory for updates.

Details

Scores and exploitation

EPSS: 0.5%Estimated probability that this CVE is exploited in the next 30 days, from the FIRST Exploit Prediction Scoring System. Read on November 6, 2026.
Disclosure date: May 12, 2026

Weaknesses and techniques

AI threat technique (MITRE ATLAS): AML.T0010

Classification

Attack type: Supply chain
Attack sophistication: Trivial
AI component targeted: Framework
Impact: Confidentiality,
Integrity
Labels: security

How this record was produced

Summaries, mitigations and labels on this page are generated automatically from the source. The linked source is authoritative. The source gives this advisory a description and no title, so the title above is a headline generated from the source text.

Original source: https://nvd.nist.gov/vuln/detail/CVE-2026-31214

First tracked: May 12, 2026 at 02:07 PM ET, from NVD/CVE Database

Classifier: claude-haiku-5-5, prompt v4

Classifier confidence: 90% (self-assessed)

Summary

Mitigation

Related items