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.

Cadre d'apprentissage profond (PyTorch Lightning). Organisez le code PyTorch dans LightningModules, configurez les formateurs pour multi-GPU/TPU, implémentez des pipelines de données, des rappels, la journalisation (W&B, TensorBoard), une formation distribuée (DDP, FSDP, DeepSpeed), pour une formation évolutive sur les réseaux neuronaux. Source : jackspace/claudeskillz.

Voir l'original