High-level training framework for PyTorch that abstracts boilerplate while maintaining flexibility. Includes the Trainer, LightningModule, and support for multi-GPU scaling and reproducibility. (light
skills/pytorch-lightning
Use when you have lint errors, formatting issues, or before committing code to ensure it passes CI.
This skill should be used when the user asks to "update documentation for my changes", "check docs for this PR", "what d...
Write docstrings for PyTorch functions and methods following PyTorch conventions. Use when writing or updating docstring...