pytorch-lightning

SKILL.md

PyTorch Lightning is a deep learning framework that organizes PyTorch code to eliminate boilerplate while maintaining full flexibility. Automate training workflows, multi-device orchestration, and implement best practices for neural network training and scaling across multiple GPUs/TPUs.

Quick template reference: See scripts/templatelightningmodule.py for a complete boilerplate.

Detailed documentation: Read references/lightningmodule.md for comprehensive method documentation, hooks, properties, and best practices.

Deep learning framework (PyTorch Lightning). Organize PyTorch code into LightningModules, configure Trainers for multi-GPU/TPU, implement data pipelines, callbacks, logging (W&B, TensorBoard), distributed training (DDP, FSDP, DeepSpeed), for scalable neural network training. Source: ovachiever/droid-tings.

View raw