deep-learning · pluginagentmarketplace — Claude Code skill: PyTorch and transformers
A deep learning skill that feeds your agent PyTorch training loops, transformer blocks and ONNX export patterns for production work.
npx skills add pluginagentmarketplace/custom-plugin-data-engineer --skill deep-learning -a claude-codeWhat deep-learning does
The skill hands the agent working Python instead of vague advice. You get a training loop with mixed precision, gradient accumulation and gradient clipping, plus a learning rate finder and an early stopping class. Architecture examples cover residual blocks and multi-head self-attention.
Deployment is part of it too: export to ONNX with torch.onnx.export, run inference through ONNX Runtime, or script the model with torch.jit.script. The tool table lists PyTorch 2.2+, PyTorch Lightning 2.2+, Hugging Face 4.38+, ONNX Runtime 1.17+, TensorRT 8.6+, Weights & Biases and Ray 2.9+.
It is marked advanced, and the prerequisites are python-programming and machine-learning. So if you have never written a PyTorch loop, this is not the place to start. The skill also unlocks llms-generative-ai and mlops.
What it helps with
- write a PyTorch training loop with mixed precision
- build a transformer for text classification
- add residual connections to a CNN
- export a trained model to ONNX
- set up gradient accumulation and early stopping
Example prompts for the agent
- Write a PyTorch training loop with mixed precision and gradient accumulation
- Build a transformer for text classification and add early stopping
- Export this model to ONNX and show me how to run it with ONNX Runtime
How to install deep-learning
1Skills CLI (Vercel)
npx skills add pluginagentmarketplace/custom-plugin-data-engineer --skill deep-learning -a claude-code2Manually: put the files in place
~/.claude/skills/deep-learning/SKILL.mdCopy the whole skill folder; for one project use .claude/skills/deep-learning/
1Skills CLI (Vercel)
npx skills add pluginagentmarketplace/custom-plugin-data-engineer --skill deep-learning -a codex2Manually: put the files in place
.agents/skills/deep-learning/SKILL.mdFor all projects use ~/.agents/skills/
1Skills CLI (Vercel)
npx skills add pluginagentmarketplace/custom-plugin-data-engineer --skill deep-learning -a cursor2Manually: put the files in place
.cursor/skills/deep-learning/SKILL.mdCursor also reads .agents/skills/ and .claude/skills/
1GitHub CLI
gh skill install pluginagentmarketplace/custom-plugin-data-engineer deep-learning2Manually: put the files in place
.github/skills/deep-learning/SKILL.md1Skills CLI (Vercel)
npx skills add pluginagentmarketplace/custom-plugin-data-engineer --skill deep-learning -a gemini-cli2Platform CLI
gemini skills install https://github.com/pluginagentmarketplace/custom-plugin-data-engineer --path skills/deep-learningManually: put the files in place
.devin/skills/deep-learning/SKILL.mdThe legacy .windsurf/skills/ path still works
Skills CLI (Vercel)
npx skills add pluginagentmarketplace/custom-plugin-data-engineer --skill deep-learning -a opencodeClaude.ai
Zip the folder and upload it in Settings → Capabilities → Skills
Step by step, with commands for each platform: How to install skills in Claude Code: five ways
Questions about deep-learning
How do I install deep-learning in Claude?
Run npx skills add pluginagentmarketplace/custom-plugin-data-engineer --skill deep-learning -a claude-code. The skill lands in ~/.claude/skills/deep-learning/SKILL.md.
What does the deep-learning skill do?
It guides the agent through deep learning code: training loops, network architectures and model export. Built around PyTorch, ONNX Runtime and Hugging Face.
What should I know before using it?
The prerequisites are python-programming and machine-learning, and the skill is rated advanced.
Similar skills
Everything for AI and ML →deep-learning by other authors comes first.
