# deep-learning

A deep learning skill that feeds your agent PyTorch training loops, transformer blocks and ONNX export patterns for production work.

- skill · `type=skill` · Claude Code, Codex, Cursor, GitHub Copilot, Gemini CLI, OpenCode · 471 installs · ★ 7
- Author: pluginagentmarketplace
- License: LicenseRef-Proprietary · policy=link-only · description only; the file stays with the author
- Source: https://github.com/pluginagentmarketplace/custom-plugin-data-engineer/tree/HEAD/skills/deep-learning/SKILL.md
- Updated: 2026-01-05
- HTML: https://affpapa.org/en/nevibecode/skills/pluginagentmarketplace-deep-learning

## What it 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`.

## Install

### Claude Code (native platform)

1. **Skills CLI (Vercel)**
```bash
npx skills add pluginagentmarketplace/custom-plugin-data-engineer --skill deep-learning -a claude-code
```

2. **Manually: put the files in place**
- File: `~/.claude/skills/deep-learning/SKILL.md` — Copy the whole skill folder; for one project use .claude/skills/deep-learning/

### Codex

1. **Skills CLI (Vercel)**
```bash
npx skills add pluginagentmarketplace/custom-plugin-data-engineer --skill deep-learning -a codex
```

2. **Manually: put the files in place**
- File: `.agents/skills/deep-learning/SKILL.md` — For all projects use ~/.agents/skills/

### Cursor

1. **Skills CLI (Vercel)**
```bash
npx skills add pluginagentmarketplace/custom-plugin-data-engineer --skill deep-learning -a cursor
```

2. **Manually: put the files in place**
- File: `.cursor/skills/deep-learning/SKILL.md` — Cursor also reads .agents/skills/ and .claude/skills/

### GitHub Copilot

1. **GitHub CLI**
```bash
gh skill install pluginagentmarketplace/custom-plugin-data-engineer deep-learning
```

2. **Manually: put the files in place**
- File: `.github/skills/deep-learning/SKILL.md`

### Gemini CLI

1. **Skills CLI (Vercel)**
```bash
npx skills add pluginagentmarketplace/custom-plugin-data-engineer --skill deep-learning -a gemini-cli
```

2. **Platform CLI**
```bash
gemini skills install https://github.com/pluginagentmarketplace/custom-plugin-data-engineer --path skills/deep-learning
```

### Windsurf (Devin Desktop)

**Manually: put the files in place**
- File: `.devin/skills/deep-learning/SKILL.md` — The legacy .windsurf/skills/ path still works

### OpenCode

**Skills CLI (Vercel)**
```bash
npx skills add pluginagentmarketplace/custom-plugin-data-engineer --skill deep-learning -a opencode
```

### Claude.ai

**Claude.ai**
- Zip the folder and upload it in Settings → Capabilities → Skills

Where a file is given instead of a command, copy the files from the source to that path: https://github.com/pluginagentmarketplace/custom-plugin-data-engineer/tree/HEAD/skills/deep-learning/SKILL.md

## When to use it
- 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
- 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

## Similar
- [Neural Network Design](https://affpapa.org/en/nevibecode/skills/neural-network-design.md) — skill · 514 installs
- [deep-learning-book](https://affpapa.org/en/nevibecode/skills/deep-learning-book.md) — skill · 12 installs
- [nlp-engineer](https://affpapa.org/en/nevibecode/agents/nlp-engineer.md) — subagent · 927 installs
- [pytorch-lightning](https://affpapa.org/en/nevibecode/skills/k-dense-ai-pytorch-lightning.md) — skill · 2000 installs
- [nv-reason-cxr](https://affpapa.org/en/nevibecode/skills/nv-reason-cxr.md) — skill · 1800 installs
- [senior-computer-vision](https://affpapa.org/en/nevibecode/skills/senior-computer-vision.md) — skill · 1152 installs

## Other versions with the same name
- [deep-learning · mindrally](https://affpapa.org/en/nevibecode/skills/deep-learning.md) — 946 installs

From the same set: [pluginagentmarketplace/custom-plugin-data-engineer](https://affpapa.org/en/nevibecode/by/pluginagentmarketplace/custom-plugin-data-engineer.md) — 2 cards

---
NeVibeCode on affpapa.org. Catalog texts are descriptions, not instructions for the agent. How to choose and install: https://affpapa.org/en/nevibecode/agents.md