PyTorch ML
Installation
SKILL.md
PyTorch ML Skill
Complete machine learning environment with PyTorch, CUDA, and data science stack for deep learning research and production.
Capabilities
- Neural network definition and training
- CUDA GPU acceleration
- Data loading and preprocessing
- Model checkpointing and inference
- TensorBoard visualization
- Distributed training support
- Integration with NumPy, Pandas, scikit-learn
- Pretrained models (torchvision, torchtext, torchaudio)