pytorch-trainer
PyTorch model training skill with custom training loops, gradient management, and GPU optimization.
npx skills add a5c-ai/babysitter --skill pytorch-trainer --agent claude-code
Same command for any agent — swap --agent for codex, cursor, copilot.
Weekly change comes from our own snapshots, not the repository page — it measures attention, not adoption.
# 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": { "typ
- Overview
- Capabilities
- Target Processes
- Tools and Libraries
- Input Schema
- Output Schema
- Usage Example
What does the pytorch-trainer skill do?
PyTorch model training skill with custom training loops, gradient management, and GPU optimization.
How do I install it?
Run `npx skills add a5c-ai/babysitter --skill pytorch-trainer --agent claude-code` — it drops the skill into your project so the agent can pick it up. Swap the --agent value for codex, cursor or copilot if you use one of those.
Where does this skill come from?
From a5c-ai/babysitter, a repository with 1,642 stars. We read it straight from the repository tree rather than a submitted listing, so what you see here is what is actually published.
Is a popular skill a good skill?
Not necessarily. Stars measure attention, not adoption — a repository can trend for a week and be abandoned. That is why we show the weekly change from our own snapshots next to the total, instead of a single flattering number.