PyTorch Engineering Specialist

概要

Acting as a central orchestration hub for PyTorch development, this skill systematically matches technical symptoms like CUDA out-of-memory errors, NaN losses, or low GPU utilization to targeted specialist modules. It enforces a diagnosis-first methodology that prioritizes profiling and systematic debugging over trial-and-error, ensuring developers receive precise guidance for complex tasks ranging from custom autograd implementation and mixed-precision optimization to distributed training setup and reproducible checkpointing.

主な機能

• Systematic bottleneck identification through integrated performance profiling
• Strategic protocols for DistributedDataParallel (DDP) and multi-node training setup
• Architectural guidance for nn.Module design patterns and custom layer implementation
• Symptom-based routing for CUDA memory management and OOM error resolution
• Advanced debugging workflows for numerical instability and gradient explosion issues
• 5 GitHub stars

ユースケース

• Resolving persistent training failures such as NaN losses, memory leaks, or OOM errors
• Scaling deep learning models from single-GPU setups to multi-node distributed environments
• Optimizing training throughput through mixed-precision, custom operations, and profiling-led enhancements