Agent skill

Pytorch Trainer

by majiayu000 in majiayu000/claude-skill-registry

PyTorch model training skill with custom training loops, gradient management, and GPU optimization.

MITAuto-check: notesAI & LLM Engineering

Install Pytorch Trainer

skills CLI

$ npx skills add majiayu000/claude-skill-registry --skill pytorch-trainer -a claude-code

Project install by default; add -g for ~/.claude/skills/.

GitHub CLI

$ gh skill install majiayu000/claude-skill-registry pytorch-trainer --agent claude-code

Project scope by default; add --scope user for a personal install. Needs GitHub CLI 2.90.0 or later (public preview).

Manual copy

$ git clone --depth 1 https://github.com/majiayu000/claude-skill-registry.git skills-src && mkdir -p .claude/skills && cp -r skills-src/skills/ai-ml/pytorch-trainer .claude/skills/pytorch-trainer && rm -rf skills-src

Use ~/.claude/skills/ instead of .claude/skills for a personal install. The folder must contain SKILL.md.

Claude Code skills documentation · loads skills from .claude/skills/

Paste into Claude Code

Install the "pytorch-trainer" agent skill from https://github.com/majiayu000/claude-skill-registry/tree/main/skills/ai-ml/pytorch-trainer into .claude/skills/pytorch-trainer/ in this project. Copy the whole folder (SKILL.md and every file beside it), keep the folder name "pytorch-trainer", then confirm the skill loads.

Claude Code copies the folder itself, the same result as the manual copy. Check what it changed before you commit it.

$skill-installer

$skill-installer install https://github.com/majiayu000/claude-skill-registry/tree/main/skills/ai-ml/pytorch-trainer

Type this inside Codex. $skill-installer <name> installs a curated skill from openai/skills. The installer writes to $CODEX_HOME/skills (default ~/.codex/skills). Restart Codex if the skill does not show up.

skills CLI

$ npx skills add majiayu000/claude-skill-registry --skill pytorch-trainer -a codex

Project install goes to .agents/skills/; add -g for ~/.codex/skills/.

GitHub CLI

$ gh skill install majiayu000/claude-skill-registry pytorch-trainer --agent codex

Project scope by default (.agents/skills/); add --scope user for a personal install.

Manual copy

$ git clone --depth 1 https://github.com/majiayu000/claude-skill-registry.git skills-src && mkdir -p .agents/skills && cp -r skills-src/skills/ai-ml/pytorch-trainer .agents/skills/pytorch-trainer && rm -rf skills-src

Use ~/.agents/skills/ instead of .agents/skills for a personal install.

Codex skills documentation · loads skills from .agents/skills/

Paste into Codex

Install the "pytorch-trainer" agent skill from https://github.com/majiayu000/claude-skill-registry/tree/main/skills/ai-ml/pytorch-trainer into .agents/skills/pytorch-trainer/ in this project. Copy the whole folder (SKILL.md and every file beside it), keep the folder name "pytorch-trainer", then confirm the skill loads.

Codex copies the folder itself, the same result as the manual copy. Check what it changed before you commit it.

skills CLI

$ npx skills add majiayu000/claude-skill-registry --skill pytorch-trainer -a cursor

Project install goes to .agents/skills/; add -g for ~/.cursor/skills/.

GitHub CLI

$ gh skill install majiayu000/claude-skill-registry pytorch-trainer --agent cursor

Project scope by default (.agents/skills/); add --scope user for a personal install.

Manual copy

$ git clone --depth 1 https://github.com/majiayu000/claude-skill-registry.git skills-src && mkdir -p .cursor/skills && cp -r skills-src/skills/ai-ml/pytorch-trainer .cursor/skills/pytorch-trainer && rm -rf skills-src

Use ~/.cursor/skills/ instead of .cursor/skills for a personal install.

Cursor skills documentation · loads skills from .cursor/skills/, .agents/skills/, .claude/skills/, .codex/skills/

Paste into Cursor

Install the "pytorch-trainer" agent skill from https://github.com/majiayu000/claude-skill-registry/tree/main/skills/ai-ml/pytorch-trainer into .cursor/skills/pytorch-trainer/ in this project. Copy the whole folder (SKILL.md and every file beside it), keep the folder name "pytorch-trainer", then confirm the skill loads.

Cursor copies the folder itself, the same result as the manual copy. Check what it changed before you commit it.

gemini skills install

$ gemini skills install https://github.com/majiayu000/claude-skill-registry.git --path skills/ai-ml/pytorch-trainer

--scope user (default) or --scope workspace; --path is the subfolder of the repo that holds the skill; --consent skips the security confirmation prompt.

skills CLI

$ npx skills add majiayu000/claude-skill-registry --skill pytorch-trainer -a gemini-cli

Project install goes to .agents/skills/; add -g for ~/.gemini/skills/.

GitHub CLI

$ gh skill install majiayu000/claude-skill-registry pytorch-trainer --agent gemini-cli

Project scope by default (.agents/skills/); add --scope user for a personal install.

Manual copy

$ git clone --depth 1 https://github.com/majiayu000/claude-skill-registry.git skills-src && mkdir -p .gemini/skills && cp -r skills-src/skills/ai-ml/pytorch-trainer .gemini/skills/pytorch-trainer && rm -rf skills-src

Use ~/.gemini/skills/ instead of .gemini/skills for a personal install, then run /skills reload.

Gemini CLI skills documentation · loads skills from .gemini/skills/, .agents/skills/

Paste into Gemini CLI

Install the "pytorch-trainer" agent skill from https://github.com/majiayu000/claude-skill-registry/tree/main/skills/ai-ml/pytorch-trainer into .gemini/skills/pytorch-trainer/ in this project. Copy the whole folder (SKILL.md and every file beside it), keep the folder name "pytorch-trainer", then confirm the skill loads.

Gemini CLI copies the folder itself, the same result as the manual copy. Check what it changed before you commit it.

GitHub CLI

$ gh skill install majiayu000/claude-skill-registry pytorch-trainer

Installs for Copilot at project scope by default; add --scope user for a personal install. Preview a skill first with gh skill preview. Needs GitHub CLI 2.90.0 or later (public preview).

skills CLI

$ npx skills add majiayu000/claude-skill-registry --skill pytorch-trainer -a github-copilot

Project install goes to .agents/skills/; add -g for ~/.copilot/skills/.

Manual copy

$ git clone --depth 1 https://github.com/majiayu000/claude-skill-registry.git skills-src && mkdir -p .github/skills && cp -r skills-src/skills/ai-ml/pytorch-trainer .github/skills/pytorch-trainer && rm -rf skills-src

Use ~/.copilot/skills/ instead of .github/skills for a personal install. Commit .github/skills so cloud agent and code review can use it.

GitHub Copilot skills documentation · loads skills from .github/skills/, .claude/skills/, .agents/skills/

Paste into GitHub Copilot

Install the "pytorch-trainer" agent skill from https://github.com/majiayu000/claude-skill-registry/tree/main/skills/ai-ml/pytorch-trainer into .github/skills/pytorch-trainer/ in this project. Copy the whole folder (SKILL.md and every file beside it), keep the folder name "pytorch-trainer", then confirm the skill loads.

GitHub Copilot copies the folder itself, the same result as the manual copy. Check what it changed before you commit it.

skills CLI

$ npx skills add majiayu000/claude-skill-registry --skill pytorch-trainer -a opencode

OpenCode documents no install command of its own. Project install goes to .agents/skills/; add -g for ~/.config/opencode/skills/.

GitHub CLI

$ gh skill install majiayu000/claude-skill-registry pytorch-trainer --agent opencode

Project scope by default (.agents/skills/); add --scope user for a personal install.

Manual copy

$ git clone --depth 1 https://github.com/majiayu000/claude-skill-registry.git skills-src && mkdir -p .opencode/skills && cp -r skills-src/skills/ai-ml/pytorch-trainer .opencode/skills/pytorch-trainer && rm -rf skills-src

Use ~/.config/opencode/skills/ instead of .opencode/skills for a personal install.

OpenCode skills documentation · loads skills from .opencode/skills/, .claude/skills/, .agents/skills/

Paste into OpenCode

Install the "pytorch-trainer" agent skill from https://github.com/majiayu000/claude-skill-registry/tree/main/skills/ai-ml/pytorch-trainer into .opencode/skills/pytorch-trainer/ in this project. Copy the whole folder (SKILL.md and every file beside it), keep the folder name "pytorch-trainer", then confirm the skill loads.

OpenCode copies the folder itself, the same result as the manual copy. Check what it changed before you commit it.

Facts

Skill name: pytorch-trainer
GitHub stars: 666
Used in: 1 other repo
Token cost: ~884 tokens
SKILL.md length: 86 words
Files: 2
Skills in repo: 1,273
Repo updated: Oct 7, 2026
First seen: Oct 7, 2026
Licence: MIT

At a glance

PyTorch model training skill with custom training loops, gradient management, and GPU optimization.

Tasks that involve Deep learning
SKILL.md covers Overview, Capabilities, Target Processes and Tools and Libraries, plus 3 more sections
Instructions only: no scripts, shell commands, URLs or credentials in SKILL.md

What it does

Pytorch Trainer is an agent skill from majiayu000/claude-skill-registry. PyTorch model training skill with custom training loops, gradient management, and GPU optimization.

Its SKILL.md is about 880 tokens, which your agent loads only when the skill is triggered. The skill folder holds 1 other file (for example `metadata.json`).

It sits in AI & LLM Engineering, covering Deep learning. It works with PyTorch. The repository describes itself as: Searchable Claude Code skills catalog with source-linked guides and generated registry artifacts. The licence is MIT.

When your agent uses it

Tasks that involve Deep learning

Example prompts

“/pytorch-trainer”

Requirements

Pre-approved tools (allowed-tools): Read, Write, Bash, Glob, Grep

What it can do on your machine

Read from SKILL.md and the folder at commit 2d14a69. It shows what the files ask for, not the result of running them.

Tool permissions
Pre-approves these tools, so the agent can use them without asking each time:
- Read
- Write
- Bash
- Glob
- Grep
From allowed-tools in the SKILL.md frontmatter.
Runs code
No scripts in the folder and no shell commands in SKILL.md (its code samples are json and javascript).
From the folder's file list and the shell code blocks in SKILL.md.
Network
No URLs in SKILL.md.
From URLs in SKILL.md, links to its own repository left out.
Credentials
Names no API keys, tokens, secrets or passwords.
From names ending in _API_KEY, _TOKEN, _SECRET, _KEY or _PASSWORD in SKILL.md.

Context cost

Pytorch Trainer loads about 884 tokens when it runs. Until then it costs about 29 tokens; SKILL.md has 86 words of instructions outside code blocks.

Always · name and description, kept in context so the agent knows when to use it: ~29
When it runs · the whole SKILL.md, loaded when a task matches: ~884

Estimates: characters ÷ 4, the usual rule of thumb; real counts depend on the model's tokenizer. Scripts and assets cost tokens only if the agent reads them.

Safety

Auto-check: notes

The automated check noted patterns worth knowing about, such as sudo or a known installer.

NotePre-approves every shell command (allowed-tools: Bash)SKILL.md
```
allowed-tools: Read, Write, Bash, Glob, Grep
```

Automated static check — not a guarantee. Review scripts before installing. It scans the text of SKILL.md for risky patterns (piping downloads into a shell, reading credential files, hidden Unicode, destructive commands); files beside SKILL.md are not scanned.

SKILL.md

The full file from majiayu000/claude-skill-registry at commit 2d14a69, republished under its MIT licence (© majiayu000). 86 words, ~884 tokens.

Download SKILL.mdSave it as .claude/skills/pytorch-trainer/SKILL.md (or your agent's skills folder). This skill also uses 1 other file; get the full folder from GitHub.

name: pytorch-trainer
description: PyTorch model training skill with custom training loops, gradient management, and GPU optimization.
allowed-tools: Read, Write, Bash, Glob, Grep

pytorch-trainer

Overview

PyTorch model training skill with custom training loops, gradient management, GPU optimization, and integration with experiment tracking systems.

Capabilities

Custom training loop execution
Learning rate scheduling (StepLR, CosineAnnealing, OneCycleLR, etc.)
Gradient clipping and accumulation
Mixed precision training (AMP)
Checkpoint management and resumption
DataLoader optimization
Multi-GPU training (DataParallel, DistributedDataParallel)
Early stopping with patience

Target Processes

Model Training Pipeline with Experiment Tracking
Distributed Training Orchestration
AutoML Pipeline Orchestration

Tools and Libraries

PyTorch
PyTorch Lightning (optional)
torchvision, torchaudio, torchtext
CUDA toolkit

Input Schema

json

{
  "type": "object",
  "required": ["modelPath", "dataConfig", "trainingConfig"],
  "properties": {
    "modelPath": {
      "type": "string",
      "description": "Path to model definition file"
    },
    "dataConfig": {
      "type": "object",
      "properties": {
        "trainPath": { "type": "string" },
        "valPath": { "type": "string" },
        "batchSize": { "type": "integer" },
        "numWorkers": { "type": "integer" }
      }
    },
    "trainingConfig": {
      "type": "object",
      "properties": {
        "epochs": { "type": "integer" },
        "learningRate": { "type": "number" },
        "optimizer": { "type": "string" },
        "scheduler": { "type": "string" },
        "mixedPrecision": { "type": "boolean" },
        "gradientClipping": { "type": "number" },
        "gradientAccumulation": { "type": "integer" }
      }
    },
    "checkpointConfig": {
      "type": "object",
      "properties": {
        "saveDir": { "type": "string" },
        "saveEvery": { "type": "integer" },
        "resumeFrom": { "type": "string" }
      }
    }
  }
}

Output Schema

json

{
  "type": "object",
  "required": ["status", "metrics", "checkpointPath"],
  "properties": {
    "status": {
      "type": "string",
      "enum": ["success", "error", "early_stopped"]
    },
    "metrics": {
      "type": "object",
      "properties": {
        "trainLoss": { "type": "number" },
        "valLoss": { "type": "number" },
        "trainAccuracy": { "type": "number" },
        "valAccuracy": { "type": "number" },
        "epochsTrained": { "type": "integer" },
        "trainingTime": { "type": "number" }
      }
    },
    "checkpointPath": {
      "type": "string"
    },
    "learningCurve": {
      "type": "array",
      "items": {
        "type": "object",
        "properties": {
          "epoch": { "type": "integer" },
          "trainLoss": { "type": "number" },
          "valLoss": { "type": "number" }
        }
      }
    }
  }
}

Usage Example

javascript

{
  kind: 'skill',
  title: 'Train PyTorch model',
  skill: {
    name: 'pytorch-trainer',
    context: {
      modelPath: 'models/resnet.py',
      dataConfig: {
        trainPath: 'data/train',
        valPath: 'data/val',
        batchSize: 32,
        numWorkers: 4
      },
      trainingConfig: {
        epochs: 100,
        learningRate: 0.001,
        optimizer: 'AdamW',
        scheduler: 'cosine',
        mixedPrecision: true,
        gradientClipping: 1.0
      }
    }
  }
}

Files

SKILL.md and 1 other file in skills/ai-ml/pytorch-trainer of majiayu000/claude-skill-registry.

SKILL.md
metadata.json

Open the folder on GitHubat commit 2d14a69

Used in 1 other repository

We found 1 copy of this SKILL.md (exact, near-identical or edited) in other folders, from 1 other GitHub owner. This page covers the copy in majiayu000/claude-skill-registry, which our catalogue first saw on October 7, 2026.

a5c-ai/babysitterlibrary/specializations/data-science-ml/skills/pytorch-trainer

Compare with similar skills

Pytorch Trainer next to the 5 skills that share the most tags, products or categories with it. Stars are the repository's; “used in” counts other GitHub owners with a copy.

Pytorch Trainer compared with similar skills
Skill	Stars	Used in	Tokens	Auto-check	Licence	Repo updated
Pytorch Trainer this skillmajiayu000/claude-skill-registry	666	1 repos	~884	Automated safety check: Notes	MIT	yesterday
Add Uint Supportpytorch/pytorch	104k	2 repos	~2.3k	Automated safety check: Pass	Custom licence	yesterday
CLIP Image-Text MatchingOrchestra-Research/AI-Research-SKILLs	13k	8 repos	~1.7k	Automated safety check: Pass	MIT	3 mo ago
Add Torch Shapes Examplefacebook/pyrefly	7.1k	—	~1.3k	Automated safety check: Pass	MIT	yesterday
Interview Cheatsheetwanshuiyin/ARIS-in-AI-Offer	580	1 repos	~3.4k	Automated safety check: Notes	MIT	2 days ago
Ghstack CIpytorch/pytorch	104k	—	~1.4k	Automated safety check: Pass	Custom licence	yesterday

Similar skills

Add Uint Support
pytorch/pytorch
Add unsigned integer (uint) type support to PyTorch operators by updating ATDISPATCH macros.
104k GitHub starsUsed in 2 repos~2.3k tokens
AI & LLM EngineeringAuto-check passed
CLIP Image-Text Matching
Orchestra-Research/AI-Research-SKILLs
Explains OpenAI's CLIP model for zero-shot image classification, image-text similarity, semantic image search and content moderation, with install steps and code patterns.
13k GitHub starsUsed in 8 repos~1.7k tokens
AI & LLM EngineeringAuto-check passed
Add Torch Shapes Example
facebook/pyrefly
Official
A skill your agent uses when adding a new PyTorch model to Pyrefly's shape-tracking example corpus under tensor-shapes/pyrefly-torch-stubs/examples — i.e.
7.1k GitHub stars~1.3k tokensUpdated yesterday
AI & LLM EngineeringAuto-check passed
Interview Cheatsheet
wanshuiyin/ARIS-in-AI-Offer
Generate a long-form Chinese interview-prep cheat sheet on a specific ML/LLM topic — formulas with derivations, from-scratch PyTorch code, comparison tables, and 25 高频面试题 (L1 必会 / L2 进阶 / L3 顶级 lab).
580 GitHub starsUsed in 1 repo~3.4k tokens
AI & LLM EngineeringAuto-check: notes
Ghstack CI
pytorch/pytorch
Manage CI for PyTorch ghstack stacks by running CI where its results are useful now and deferring other PRs with [no-ci].
104k GitHub stars~1.4k tokensUpdated yesterday
AI & LLM EngineeringAuto-check passed
MUSA GPU Training Optimizer
open-infra-skills/infra-skills
Profiles, benchmarks and tunes AI training workloads on Moore Threads MUSA GPUs with a measurement-first process that keeps model behavior unchanged.
141 GitHub stars~1.7k tokensUpdated 3 mo ago
AI & LLM EngineeringAuto-check passed

Questions about Pytorch Trainer

What does Pytorch Trainer do?

PyTorch model training skill with custom training loops, gradient management, and GPU optimization. Pytorch Trainer is an agent skill from majiayu000/claude-skill-registry. PyTorch model training skill with custom training loops, gradient management, and GPU optimization.

When should I use Pytorch Trainer?

Pytorch Trainer fits situations like: tasks that involve Deep learning.

How do I install Pytorch Trainer in Claude Code?

Run `npx skills add majiayu000/claude-skill-registry --skill pytorch-trainer -a claude-code`. Or copy the skill folder (skills/ai-ml/pytorch-trainer in majiayu000/claude-skill-registry) into .claude/skills/pytorch-trainer in your project. Claude Code loads it when a task matches its description.

How do I install Pytorch Trainer in Codex?

Run `npx skills add majiayu000/claude-skill-registry --skill pytorch-trainer -a codex`. Or copy the skill folder (skills/ai-ml/pytorch-trainer in majiayu000/claude-skill-registry) into .agents/skills/pytorch-trainer in your project. Codex loads it when a task matches its description.

Can I use Pytorch Trainer in Cursor, Gemini CLI or GitHub Copilot?

Cursor, Gemini CLI, GitHub Copilot and OpenCode also load SKILL.md folders. With the skills CLI, run `npx skills add majiayu000/claude-skill-registry --skill pytorch-trainer -a cursor` (or -a gemini-cli, github-copilot or opencode for the others). To copy it by hand, put the folder in .cursor/skills/pytorch-trainer, .gemini/skills/pytorch-trainer, .github/skills/pytorch-trainer and .opencode/skills/pytorch-trainer in your project.

What does Pytorch Trainer need to run?

SKILL.md names no scripts, command-line tools or credentials: Pytorch Trainer is instructions for the agent only. Its frontmatter pre-approves these tools: Read, Write, Bash, Glob, Grep.

Does Pytorch Trainer access the network?

SKILL.md contains no URLs. Any network use would come from the scripts or tools the agent runs. This is read from the text; nothing was executed.

Is Pytorch Trainer safe to install?

Our automated static check of SKILL.md found notes only (pre-approves every shell command (allowed-tools: bash)), nothing it rates as a warning. It is not a guarantee. Review the folder before installing.

What licence does Pytorch Trainer use?

Pytorch Trainer is published under the MIT licence (the repository's licence). It allows redistribution, so the full SKILL.md is shown on this page.

How many tokens does Pytorch Trainer use?

About 884 tokens (SKILL.md is roughly 3.5k characters). Agents keep only the skill's name and description in context until a task matches; then they load SKILL.md in full.

What are the alternatives to Pytorch Trainer?

Skills that share tags, products or a category with Pytorch Trainer: Add Uint Support (pytorch/pytorch, 104k stars), CLIP Image-Text Matching (Orchestra-Research/AI-Research-SKILLs, 13k stars), Add Torch Shapes Example (facebook/pyrefly, 7.1k stars) and Interview Cheatsheet (wanshuiyin/ARIS-in-AI-Offer, 580 stars). The comparison table on this page puts their stars, adoption, token cost, safety result and licence side by side.

Who maintains Pytorch Trainer?

majiayu000 (a GitHub user) maintains it in majiayu000/claude-skill-registry, which has 666 GitHub stars. The repository holds 1,273 skills in this directory. The repository was last updated on October 7, 2026.

Source: majiayu000/claude-skill-registry on GitHub. Facts on this page come from the repository at the commit we read; the author's words are quoted as theirs.

Install Pytorch Trainer

Facts

At a glance

What it does

When your agent uses it

Example prompts

Requirements

What it can do on your machine

Tool permissions

Runs code

Network

Credentials

Context cost

Safety

SKILL.md

pytorch-trainer

Overview

Capabilities

Target Processes

Tools and Libraries

Input Schema

Output Schema

Usage Example

Files

Used in 1 other repository

Compare with similar skills

Similar skills

More from majiayu000/claude-skill-registry

Related

Questions about Pytorch Trainer