refactor:pytorch
CommunityRefactor PyTorch for maintainable, fast code.
AuthorSnakeO
Version1.0.0
Installs0
System Documentation
What problem does it solve?
This Skill addresses the difficulty of maintaining large PyTorch codebases by encouraging modular design, consistent patterns, and performance-oriented practices.
Core Features & Use Cases
- Modular nn.Module blocks and utilities to avoid duplication.
- Performance enhancements using torch.compile, AMP, and optimized DataLoader.
- Structured training and data handling with clear separation of concerns and potential Lightning integration.
Quick Start
Refactor your PyTorch training script to adopt a modular nn.Module design, enable torch.compile, and switch to AMP during training.
Dependency Matrix
Required Modules
None requiredComponents
Standard package💻 Claude Code Installation
Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.
Please help me install this Skill: Name: refactor:pytorch Download link: https://github.com/SnakeO/claude-debug-and-refactor-skills-plugin/archive/main.zip#refactor-pytorch Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
Agent Skills Search Helper
Install a tiny helper to your Agent, search and equip skill from 223,000+ vetted skills library on demand.