ai/pytorch/SKILL.md
PyTorch model inspection and checkpoint workflow for loading tensors, `state_dict` data, modules, and parameters. Use when working with `.pt` or `.pth` artifacts, auditing model structure, extracting weights, or scripting inference-oriented inspection of deep-learning checkpoints.
npx skillsauth add aeondave/malskill pytorchInstall this skill globally with one command. Works with Claude Code, Cursor, and Windsurf.
3 of 9 scanners reported clean
Some scanners were skipped, did not run, or reported a non-clean status. Review each row below.
Use PyTorch when model artifacts are tensors first and everything else second.
Use PyTorch when you need to:
.pt or .pth checkpoints safely onto CPU or GPUstate_dict keys, module hierarchy, and parameter shapesimport torch
checkpoint = torch.load("model.pt", map_location="cpu")
print(type(checkpoint))
print(checkpoint.keys() if isinstance(checkpoint, dict) else "non-dict checkpoint")
state = torch.load("weights.pth", map_location="cpu")
for name, tensor in state.items():
print(name, tuple(tensor.shape), tensor.dtype)
model.eval()
for name, module in model.named_modules():
print(name, module.__class__.__name__)
for name, param in model.named_parameters():
print(name, tuple(param.shape))
model.eval()
with torch.no_grad():
output = model(sample_input)
map_location="cpu" first unless you explicitly need GPU execution.state_dict-style loading and inspection over whole-model pickle blobs when possible.weights_only=True for safer tensor-only loads from trusted workflows.eval() and torch.no_grad() belong together for stable inspection and reduced memory noise.torch.load uses pickle under the hood; do not trust untrusted checkpoint files.No bundled scripts/, references/, or assets/.
Use the official PyTorch docs for torch.load, module introspection, and checkpoint best practices.
data-ai
Scoped routing: Linux operator; hosts, sessions, users, services, packages, logs, containers, SSH, network paths, privilege evidence.
development
Offensive methodology for ICS/OT/SCADA environments in authorized industrial penetration testing and red team operations. Use when assessing PLCs, RTUs, HMIs, engineering workstations, historians, or field devices running Modbus, DNP3, EtherNet/IP, S7comm/S7+, Profinet, IEC 60870-5-104, BACnet, or OPC-UA. Covers passive OT network enumeration, protocol-level device interrogation, PLC coil/register read-write attacks, HMI session exploitation, historian and engineering workstation compromise, and safe escalation rules for critical infrastructure scope. Does not cover: general IT network exploitation (network-technique), physical hardware interfaces UART/JTAG/SPI (hardware-technique), wireless sensor network attacks (wireless-technique), RF/SDR signal analysis (hardware-ctf or wireless-technique), or CTF-framed ICS lab tasks (ics-ctf).
tools
Offensive methodology for authorized game security assessments, game client security research, and game-adjacent penetration testing in real-world engagements. Use when assessing game clients for cheating vulnerabilities, testing anti-cheat effectiveness, auditing game server protocols for score manipulation or economic fraud, reverse engineering game DRM or license validation, analyzing game save file protection, or assessing game mod/plugin security. Covers: process memory scanning and manipulation (Cheat Engine methodology), game binary reversing for license and DRM bypass, game network protocol analysis and packet replay, anti-cheat mechanism analysis, save file format reversing and tampering, speed hack and value injection techniques. Does NOT cover: CTF game challenges (game-ctf), game engine source code auditing (web-exploit-technique or vuln-search-technique for the backend), or general binary exploitation (pwn-ctf or reversing-technique).
development
Auth assessment: hardware/embedded methodology; UART/JTAG/SWD/SPI/I2C, firmware extraction, boot/debug paths, embedded OS evidence.