pytorch-knowledge-patch

Installation
SKILL.md

PyTorch Knowledge Patch

Claude's baseline knowledge covers PyTorch through 2.5. This skill provides changes from PyTorch 2.6 through 2.11 (2025-01 to 2026-03).

Quick Reference — Key API Changes

Feature API Since
Safe loading default torch.load() now weights_only=True 2.6
Compile stance control torch.compiler.set_stance("eager_on_recompile") 2.6
Custom Triton ops @torch.library.triton_op("lib::name", mutates_args=()) 2.6
Auto dynamic shapes Dim.AUTO in torch.export 2.6
Mega cache (portable) torch.compiler.save_cache_artifacts() / load_cache_artifacts() 2.7
Context parallelism context_parallel(mesh) context manager for SDPA 2.7
Foreach map torch._foreach_map(fn, tensors, ...) 2.7
Control flow ops cond, while_loop, scan, associative_scan 2.8
Hierarchical compile torch.compiler.nested_compile_region() 2.8
DCP SafeTensors dcp.FileSystemWriter(path, format="safetensors") 2.8
FSDP1 deprecated Use fully_shard() (FSDP2) instead 2.8
Related skills
Installs
3
GitHub Stars
19
First Seen
Apr 7, 2026