pytorch-lightning
Deep learning framework (PyTorch Lightning / lightning package). Organize PyTorch code into LightningModules, configure Trainers for multi-GPU/TPU, implement data pipelines, callbacks, logging (W&B, TensorBoard, MLflow), distributed training (DDP, FSDP, DeepSpeed), for scalable neural network training.
- Version
- 1.4
- License
- Apache-2.0 license
- Compatibility
- Requires Python 3.10+ with lightning 2.6.6 and compatible PyTorch. CPU examples need no network after installation. CUDA/DDP/FSDP/DeepSpeed need suitable hardware; optional loggers require separate packages and online services need credentials/network.
Pinned to revision 68105dd992f1, so it is the text this page describes rather than whatever the author pushed since.
Pre-approved tools experimental
Experimental field. Support varies between clients, so this list is what the author declared, not what your client will enforce.
- Read
- Write
- Edit
- Bash
Files
- skills/pytorch-lightning/SKILL.md
- skills/pytorch-lightning/SKILL_CN.md
- skills/pytorch-lightning/references/best_practices.md
- skills/pytorch-lightning/references/callbacks.md
- skills/pytorch-lightning/references/data_module.md
- skills/pytorch-lightning/references/distributed_training.md
- skills/pytorch-lightning/references/lightning_module.md
- skills/pytorch-lightning/references/logging.md
- skills/pytorch-lightning/references/trainer.md
- skills/pytorch-lightning/scripts/quick_trainer_setup.py
- skills/pytorch-lightning/scripts/template_datamodule.py
- skills/pytorch-lightning/scripts/template_lightning_module.py
Every link opens the file at its source, pinned to the revision this page describes.